python - 绘制 Pandas Multiindex 条形图

Question

如何将 Python Pandas 多索引数据框绘制为带有组标签的条形图？是否有任何绘图库直接支持这个？这SO post显示了使用 matplotlib 的自定义解决方案，但是否有直接支持？

举个例子:

quarter  company
Q1       Blue       100
         Green      300
Q2       Blue       200
         Green      350
Q3       Blue       300
         Green      400
Q4       Blue       400
         Green      450
Name: count, dtype: int64

...可以使用组标签绘制此数据框吗 like this ？

提前致谢

拉菲

Answer 1

import pandas as pd

data = pd.DataFrame([
        ('Q1','Blue',100),
        ('Q1','Green',300),
        ('Q2','Blue',200),
        ('Q2','Green',350),
        ('Q3','Blue',300),
        ('Q3','Green',400),
        ('Q4','Blue',400),
        ('Q4','Green',450),
    ], 
    columns=['quarter', 'company', 'value']
)
data = data.set_index(['quarter', 'company']).value

data.unstack().plot(kind='bar', stacked=True)

如果您不想堆叠条形图:

data.unstack().plot(kind='bar')