动态更新 matplotlib 中的堆积条形图

Question

我想知道如何在 matplotlib 中动态更新堆积条形图。

这个问题Dynamically updating a bar plot in matplotlib 描述了如何为普通条形图完成此操作，而不是堆积条形图。

在普通条形图中，可以通过rect.set_height(h) 完成更新，假设rects = plt.bar(range(N), x, align='center')

但是在堆积条形图中我们还需要设置底部。

p2 = plt.bar(ind, womenMeans, width, color='y',
             bottom=menMeans, yerr=menStd)

如何动态设置底部？不幸的是，“矩形”对象似乎没有“set_bottom”属性。有没有其他方法可以解决这个问题？

Answer 1

由于某种原因，您想要的 set_bottom() 函数是 set_y 在来自 bar 的返回对象中的 patches 下。根据您建议的链接，最小的示例看起来像，

import numpy as np
import matplotlib.pyplot as plt

def setup_backend(backend='TkAgg'):
    import sys
    del sys.modules['matplotlib.backends']
    del sys.modules['matplotlib.pyplot']
    import matplotlib as mpl
    mpl.use(backend)  # do this before importing pyplot
    import matplotlib.pyplot as plt
    return plt

N = 5
width = 0.35       # the width of the bars: can also be len(x) sequence

def animate():
    # http://www.scipy.org/Cookbook/Matplotlib/Animations
    mu, sigma = 100, 15
    h = mu + sigma * np.random.randn((N*2))
    p1 = plt.bar(np.arange(N), h[:N], width, color='r')
    p2 = plt.bar(np.arange(N), h[N:], width, color='b', bottom=h[:N])
    assert len(p1) == len(p2)
    maxh = 0.
    for i in range(50):
        for rect1, rect2 in zip(p1.patches, p2.patches):
            h = mu + sigma * np.random.randn(2)
            #Keep a record of maximum value of h
            maxh = max(h[0]+h[1],maxh)
            rect1.set_height(h[0])
            rect2.set_y(rect1.get_height())
            rect2.set_height(h[1])
        #Set y limits to maximum value
        ax.set_ylim((0,maxh))
        fig.canvas.draw()

plt = setup_backend()
fig, ax = plt.subplots(1,1)
win = fig.canvas.manager.window
win.after(10, animate)
plt.show()

注意，我每次迭代都使用随机数更改高度生成，因此可以压缩两个补丁数组（否则会有点混乱）。