在循环中连续渲染 numpy 数组

Question

我的代码在 while 循环中运行，并在每次迭代时返回一个新的 numpy 数组。我想将该数组连续渲染到屏幕窗口。

基于Displaying Numpy Matrix as Video、Is there a way to detach matplotlib plots so that the computation can continue?、How to update matplotlib's imshow() window interactively?和real-time plotting in while loop with matplotlib，这是我最好的尝试：

import matplotlib.pyplot as plt
import numpy as np

# plt.ion()
image = np.zeros((800, 800))
im = plt.imshow(image)
while True:
    image += .01
    im.set_data(image)
    plt.pause(0.05)
plt.show()

它不起作用（由于某种原因，它只是呈现一个不会改变颜色的紫色方块）。我在 Ubuntu 16.04 上使用 ImageMagick。感谢您的帮助。

Answer 1

简单修正：

image = np.zeros((800, 800, 3))

现在可以了。