python - 使用 pandas 按列值分组条形图

Question

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

条形图的图例必须显示 col2 中编码的值，因此 True 和 False 并且条形图应根据这些进行分组值。

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)

相反，我希望看到类似的内容:

Answer 1

我认为你需要:

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