使用 Matplotlib 绘制滚动窗口

Question

我想在while循环中将时间序列绘制为滚动窗口：图表应始终显示 10 个最近的观察结果。

我的想法是使用maxlen=10的双端队列对象并在每一步中绘制它。 令我非常惊讶的是，该情节将新值附加到旧情节； 显然它记得不再在双端队列中的值？ 为什么会这样，我该如何关闭它？

这是我正在尝试做的一个最小示例。 绘图部分基于这篇文章（尽管plt.ion()对我没有任何改变，所以我把它省略了）：

from collections import deque
import matplotlib.pyplot as plt
import numpy as np

x = 0
data = deque(maxlen=10)

while True:
    x += np.abs(np.random.randn())
    y = np.random.randn()
    data.append((x, y))

    plt.plot(*zip(*data), c='black')
    plt.pause(0.1)

我也尝试使用 Matplotlib 的动画函数来代替，但无法弄清楚如何while无限循环中做到这一点......

Answer 1

如今，使用animation模块比多次调用plt.plot更容易（并且提供更好的性能）：

from collections import deque
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import numpy as np

def animate(i):
    global x
    x += np.abs(np.random.randn())
    y = np.random.randn()
    data.append((x, y))
    ax.relim()
    ax.autoscale_view()
    line.set_data(*zip(*data))

fig, ax = plt.subplots()
x = 0
y = np.random.randn()
data = deque([(x, y)], maxlen=10)
line, = plt.plot(*zip(*data), c='black')

ani = animation.FuncAnimation(fig, animate, interval=100)
plt.show()