绘制 Pandas 多索引条形图

Posted 2023-02-18

【中文标题】绘制 Pandas 多索引条形图

【英文标题】：Plotting Pandas Multiindex Bar Chart

【发布时间】：2016-03-18 20:29:47

【问题描述】：

如何将 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绘制吗？

提前致谢，

拉菲

【问题讨论】：

下面Goodword引用的SO post解决了这个问题。

【参考方案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')

【讨论】：

谢谢@Goodword，标签也可以分组吗？喜欢this chart?

当然。只是不要传递 stacked=True 参数。试试这个：data.unstack().plot(kind='bar')

我现在看到您发送的图像中的标签与我的更正略有不同。 This SO post 应该可以帮助您解决这些标签。

感谢 @Goodword ，您引用的 SO post 成功了。