微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

python – 绘制Pandas Multiindex条形图

如何将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绘制此数据框吗?

提前致谢,

拉菲

解决方法:

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)

stacked bar chart from multiindex

如果您不想堆叠条形图:

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

not-stacked bar chart

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐