组合两个堆叠的条形图以形成一个分组的堆叠条形图

时间:2020-07-21 12:30:00

标签: python pandas matplotlib data-visualization visualization

所以我在线找到了以下代码:

import matplotlib.pyplot as plt
import matplotlib

matplotlib.style.use('ggplot')

plotdata = pd.DataFrame({
    "2018_m":[40, 12, 10, 26, 36],
    "2019_m":[19, 8, 30, 21, 38],
    "2020_m":[10, 10, 42, 17, 37]
    }, index=["Dad", "Mam", "Bro", "Sis", "Me"]
)

plotdata2 = pd.DataFrame({
    "2018_y":[20, 22, 10, 34, 12],
    "2019_y":[12, 19, 27, 35, 14],
    "2020_y":[21, 31, 52, 20, 34]
    }, index=["Dad", "Mam", "Bro", "Sis", "Me"]
)

stacked_data = plotdata.apply(lambda x: x*100/sum(x), axis=1)
stacked_data2 = plotdata2.apply(lambda x: x*100/sum(x), axis=1)
stacked_data.plot(kind="bar", stacked=True)
stacked_data2.plot(kind="bar", stacked=True)

这是输出:

enter image description here

我想知道将它们组合在一起的最佳方法是什么,以便爸爸,妈妈,兄弟等都有两个堆叠的条形?我在网上和Stack Overflow上的其他地方遇到了很多其他分组的堆叠条形码,但是它们要求您迭代定义每个条形码的值,而理想情况下,我只想引用数据框名称'plotdata '和'plotdata2'就像上面的代码一样。

谢谢!

1 个答案:

答案 0 :(得分:3)

对于两个组,您可以通过position并相应地调整宽度:

fig, ax = plt.subplots()

stacked_data.plot(kind="bar", stacked=True, width=0.4, 
                  ax=ax, position=0)
stacked_data2.plot(kind="bar", stacked=True, width=0.4, 
                   ax=ax, position=1, hatch='//')

ax.set_xlim(right=len(stacked_data)-0.5)

输出:

enter image description here