使用熊猫按栏值分组条形图

时间:2018-08-06 09:16:23

标签: python pandas matplotlib

我试图显示编码为列值的类别的分组条形图(在我的示例中为col2

条形图的图例必须显示用col2编码的值,因此TrueFalse以及条形图应根据这些值进行分组。

df = pd.DataFrame({
    'col1': ['A', 'B', 'A', 'C', 'B', 'C', 'A', 'A', 'A', 'A', 'B'],
    'col2': [True, False, True, True, False, True, False, False, False, False, True],
})
df.groupby('col1')['col2'].count().plot(kind = 'bar', legend=True)

enter image description here

相反,我想看到类似的东西:

enter image description here

1 个答案:

答案 0 :(得分:1)

我认为您需要:

df.groupby(['col1', 'col2']).agg('size').unstack().plot(kind = 'bar', legend=True)

enter image description here

相关问题