如何在熊猫中绘制条形图?

时间:2020-09-11 15:03:47

标签: python pandas plot

目标是使用Pandas内置绘图模块绘制一个如下所示堆叠的条形图

Expected Output

但是,我找不到试图实现类似目标的近距离示例。

我正在工作的示例代码如下:

import pandas as pd
import matplotlib
import matplotlib.pyplot as plt  # noqa
matplotlib.use ('TkAgg')

data = {'name': ['Reg 1', 'Reg 1', 'Reg 1','Reg 2', 'Reg 2','Reg 2','Reg 2',
                 'Reg 3','Reg 3','Reg 3','Reg 3','Reg 3','Reg 4','Reg 4',],
        'year': ['a', 'b', 'c','a','a','b','c',
                 'a','a','b','c','d','f','f'],
        'reports': ['ay', 'by', 'cy','ay','ay','ay','cy',
                 'ay','ay','by','dy','dy','ry','rx']}

df = pd.DataFrame(data)
df_occ=df.apply(pd.Series.value_counts)
ax = df_occ.plot.bar(rot=0)
plt.show ()

感谢有人可以链接到我可能会错过的适当阅读材料。 另外,我欢迎其他可以实现上述目标的图书馆。

1 个答案:

答案 0 :(得分:4)

让我们尝试一下:

fig, ax = plt.subplots(figsize=(10,6))
hatches = ['', '//']

bar_width = 0.45
names =set(df['name'])

for i, col in enumerate(['year','reports']):
    width = bar_width if i==0 else - bar_width
    s = pd.crosstab(df['name'], df[col])
    s.plot.bar(width=width, align='edge', stacked=True, ax=ax, edgecolor='w', hatch=hatches[i])
    
    for j, name in enumerate(names):
        ax.text(j + width/2, 0, col, ha='center')


    
xmin,xmax = ax.get_xlim()
ax.set_xlim(xmin-0.4, xmax+0.5)
ax.legend(loc=[1.05, 0])

输出:

enter image description here