在函数中更新 matplotlib 图像

Question

我有一个处理图像的循环，我希望在每 100 次迭代（比如）使用 matplotlib 在单个输出窗口中显示图像。所以我正在尝试编写一个函数，它将一个 numpy 张量作为输入并显示相应的图像。

这是我所拥有的不起作用的东西：

def display(image):
  global im

  # If  im has been initialized, update it with the current image; otherwise initialize im and update with current image.   
  try:
    im
    im.set_array(image)
    plt.draw()
  except NameError:
    im = plt.imshow(image, cmap=plt.get_cmap('gray'), vmin=0, vmax=255)
    plt.show(block=False)
    plt.draw()

我一开始试图通过 FuncAnimation 传递它，但这似乎旨在让动画调用一个函数来执行更新，而不是在 matplotlib 上调用函数来显示结果。

上面的代码打开一个窗口，但它似乎没有更新。任何人都可以指出我正确的方向吗？

非常感谢，

贾斯汀

Answer 1

也许您可以使用以下组合：

• fig.canvas.draw_idle() 和
• plt.pause()

第一个将重新绘制您的图形，而第二个将调用 GUI 事件循环来更新图形。

此外，您不需要一直调用 imshow，在您的“im”对象上调用“set_data”方法就足够了。这样的事情应该工作：

import matplotlib.pyplot as plt
import numpy

fig,ax = plt.subplots(1,1)
image = numpy.array([[1,1,1], [2,2,2], [3,3,3]])
im = ax.imshow(image)

while True:       
    image = numpy.multiply(1.1, image)
    im.set_data(image)
    fig.canvas.draw_idle()
    plt.pause(1)

这是改编自这个答案。希望能帮助到你。