Одним из моих любимых аспектов использования библиотеки ggplot2
в R является способность легко определять эстетику. Я могу быстро сделать диаграмму рассеяния и применить цвет, связанный с конкретным столбцом, и я бы с удовольствием мог это сделать с помощью python/pandas/matplotlib. Мне интересно, есть ли какие-либо удобные функции, которые люди используют для сопоставления цветов с значениями с использованием кадровых фреймов pandas и Matplotlib?
##ggplot scatterplot example with R dataframe, 'df', colored by col3
ggplot(data = df, aes(x=col1, y=col2, color=col3)) + geom_point()
##ideal situation with pandas dataframe, 'df', where colors are chosen by col3
df.plot(x=col1,y=col2,color=col3)
EDIT: Спасибо за ваши ответы, но я хочу включить образец данных для уточнения того, что я прошу. Два столбца содержат числовые данные, а третья - категориальная переменная. Сценарий, о котором я думаю, назначит цвета на основе этого значения.
import pandas as pd
df = pd.DataFrame({'Height':np.random.normal(10),
'Weight':np.random.normal(10),
'Gender': ["Male","Male","Male","Male","Male",
"Female","Female","Female","Female","Female"]})