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

questionAnswers(4)

yourAnswerToTheQuestion