How to plot bar stack in Pandas?

Question

The objective is to plot a bar that stack as shown below using the Pandas builtin plot module

However, I cannot find any near example that tried to achieve similar objective.

The sample code that I am working is as below:

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 ()

Appreciate if someone can link to the appropriate reading material that I might miss. Also, I am open to other library that can achieve the aforementioned objective.

Quang Hoang · Accepted Answer · 2020-09-11 17:08:37Z

4

Let's try:

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])

Output:

edited Sep 11, 2020 at 17:08

answered Sep 11, 2020 at 15:19

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

rpb Over a year ago

Hi @quang, thanks and this is almost perfect. One trivial request if you can put label 'year' and 'reports' at each bar.

Collectives™ on Stack Overflow

How to plot bar stack in Pandas?

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related