pandas - 如何将变量描述为 pandas 中的颜色？

Question

考虑一下 Titanic 数据集，我想绘制一个条形图，显示性别和乘客等级如何影响生存。我使用下面的代码对性别和乘客类别进行分组以生成附图。如何从 x 轴中消除性别变量并将其表示为颜色（例如，用蓝色阴影男性和用红色阴影女性）？

import pandas as pd
data = pd.read_csv("train.csv")
data[['Sex','Pclass','Survived']].groupby(['Sex','Pclass']).sum().plot.bar()

标签： pandaspandas-groupby

Answer 1

这应该有效：

data=data[['Sex','Pclass','Survived']].groupby(['Sex','Pclass']).sum().reset_index()
data=data.pivot(index='Pclass',columns='Sex',values='Survived')
data.plot(kind='bar',stacked=True,color={'male':'b','female':'r'})