Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Multiple stacked bar plot with pandas

I am trying to make a multiple stacked bar plot with pandas but I'm running into issues. Here is a sample code:

import pandas as pd

df = pd.DataFrame({'a':[10, 20], 'b': [15, 25], 'c': [35, 40], 'd':[45, 50]}, index=['john', 'bob'])

ax = df[['a', 'c']].plot.bar(width=0.1, stacked=True)
ax=df[['b', 'd']].plot.bar(width=0.1, stacked=True, ax=ax)
df[['a', 'd']].plot.bar(width=0.1, stacked=True, ax=ax)

Which produces the following plot:

enter image description here

As you can see, the bars within each cluster are plotted on top of each other, which is not what I want to achieve. I want the bars within the same cluster to be plotted next to each other. I tried to play with the "position" argument but without much success.

Any idea on how to achieve this?

like image 468
fireboot Avatar asked Aug 18 '16 08:08

fireboot


1 Answers

You could do it by shifting the position parameter of a bar-plot so that they are adjacent to each other as shown:

matplotlib.style.use('ggplot')

fig, ax = plt.subplots()
df[['a', 'c']].plot.bar(stacked=True, width=0.1, position=1.5, colormap="bwr", ax=ax, alpha=0.7)
df[['b', 'd']].plot.bar(stacked=True, width=0.1, position=-0.5, colormap="RdGy", ax=ax, alpha=0.7)
df[['a', 'd']].plot.bar(stacked=True, width=0.1, position=0.5, colormap="BrBG", ax=ax, alpha=0.7)
plt.legend(loc="upper center")
plt.show()

enter image description here

like image 102
Nickil Maveli Avatar answered Sep 20 '22 07:09

Nickil Maveli