plotar cores diferentes para diferentes níveis categóricos usando o matplotlib
Eu tenho esse quadro de dadosdiamonds
que é composto de variáveis como(carat, price, color)
e quero desenhar um gráfico de dispersão deprice
paracarat
para cadacolor
, o que significa diferentecolor
tem cor diferente na trama.
Isso é fácil emR
comggplot
:
ggplot(aes(x=carat, y=price, color=color), #by setting color=color, ggplot automatically draw in different colors
data=diamonds) + geom_point(stat='summary', fun.y=median)
Gostaria de saber como isso poderia ser feito em Python usandomatplotlib
?
PS:
Eu sei sobre pacotes de plotagem auxiliar, comoseaborn
eggplot for python
, e eu não os prefiro, só quero descobrir se é possível fazer o trabalho usandomatplotlib
sozinho,; P