Matplotlib — 如何将堆叠的条形图放入子图中

Question

我正在尝试在 matplotlib 中的同一图中 plot 两个堆叠条形图。 下面的代码不是绝对最小的，因为即使我将相同的 plot 调用到两个不同的子图中，我也会遇到相同的错误，但是由于没有理由 plot 相同的 Z32FA6E1B78A9D4 groups A 列两次 I2A4 包括情况比较平易近人。

df = pd.DataFrame({
   'groups': ['A', 'A', 'B', 'A', 'A', 'B', 'B', 'A', 'B', 'B'],
   'type_one': [223.3, 17.2, 13.5, 2.4, 9.2, 43.3, 15.2, 23.5, 21.4, 98.2],
   'type_two': [419.6, 42.0, 24.6, 33.6, 21.8, 11.5, 12.4, 91.2, 41.3, 16.2]
})

plt.figure(figsize=(12,4))

plt.subplot(121)
df[df.groups=='A'].plot.bar(stacked=True)

plt.subplot(122)
df[df.groups=='B'].plot.bar(stacked=True)

plt.show()

当我在 jupyter notebooks 中运行它时，我得到两个空的子图（一个在另一个之下，所以不在同一个图中），然后两个图中的第二个出现在下面。

无论我将 stack stacked=设置为True还是False都会发生这种情况，所以我猜它与我在文档中找不到的.plot.bar()有关。 当我用plt.bar()尝试同样的事情时， plot 会正常生产。

Answer 1

捕获由subplot创建的轴并传递给bar ：

ax = plt.subplot(121)
df[df.groups=='A'].plot.bar(stacked=True, ax=ax)

ax = plt.subplot(122)
df[df.groups=='B'].plot.bar(stacked=True, ax=ax)

Output：

注意您可以使用groupby简化代码：

fig, axes = plt.subplots(1,2,figsize=(12,4))

for (k,d), ax in zip(df.groupby('groups'), axes.flat):
    d.plot.bar(stacked=True, ax=ax)