python - pandas dataframe 分级数据绘图

Question

我有一个 pandas 数据框 (df)，其中 df.age 列的范围为 0 - 100 岁，df.haircolor 列(亮、暗、紫色、灰色)。

现在我想以几十年为单位对年龄进行分类:

bins = np.linspace(df.age.min(), df.age.max(), 10)
decades = df.groupby(np.digitize(df.age, bins))

现在我正在尝试找到一种绘制此图的好方法。我想要一个条形图，其中每种头发颜色都有一个条形。我天真地尝试过。

df['haircolor'].plot(kind='bar', by=decades)

它没有给我我希望的结果。有人有想法吗？谢谢。

Answer 1

试试这个:

df['decade'] = df.age // 10 * 10
df.groupby(['decade', 'haircolor']).haircolor.count().plot(kind='bar')