from plotnine import ggplot, aes, geom_count, scale_size_continuous
from plotnine.data import diamonds
Categorized Data Plot
geom_count() makes the point size proportional to the number of points at a location
diamonds.head()
carat | cut | color | clarity | depth | table | price | x | y | z | |
---|---|---|---|---|---|---|---|---|---|---|
0 | 0.23 | Ideal | E | SI2 | 61.5 | 55.0 | 326 | 3.95 | 3.98 | 2.43 |
1 | 0.21 | Premium | E | SI1 | 59.8 | 61.0 | 326 | 3.89 | 3.84 | 2.31 |
2 | 0.23 | Good | E | VS1 | 56.9 | 65.0 | 327 | 4.05 | 4.07 | 2.31 |
3 | 0.29 | Premium | I | VS2 | 62.4 | 58.0 | 334 | 4.20 | 4.23 | 2.63 |
4 | 0.31 | Good | J | SI2 | 63.3 | 58.0 | 335 | 4.34 | 4.35 | 2.75 |
(
ggplot(diamonds)+ geom_count(aes(x="cut", y="color"))
)
We can adjust the size range of the points with scale_size_continuous
(
ggplot(diamonds)+ geom_count(aes(x="cut", y="color"))
+ scale_size_continuous(range=[1, 20])
)