如何交互更新matplotlib的imshow（）窗口？

Question

我正在研究一些计算机视觉算法，我想展示一个numpy数组在每个步骤中如何变化。

现在有效的是，如果我imshow( array )在代码末尾有一个简单的代码，则该窗口将显示并显示最终图像。

但是，我想做的是在每次迭代中更改图像时更新并显示imshow窗口。

因此，例如，我想这样做：

import numpy as np
import matplotlib.pyplot as plt
import time

array = np.zeros( (100, 100), np.uint8 )

for i in xrange( 0, 100 ):
    for j in xrange( 0, 50 ):
        array[j, i] = 1

        #_show_updated_window_briefly_
        plt.imshow( array )
        time.sleep(0.1)

问题在于，只有完成整个计算后，Matplotlib窗口才会被激活。

我已经尝试了原生matplotlib和pyplot，但是结果是一样的。为了绘制命令，我找到了一个.ion()开关，但是在这里似乎不起作用。

Q1。连续显示对numpy数组（实际上是uint8灰度图像）的更新的最佳方法是什么？

Q2。是否可以通过动画功能（例如动态图像示例）来执行此操作？我想在循环内调用一个函数，因此我不知道如何使用动画函数来实现这一点。

Answer 1

您无需一直打来电话imshow。使用对象的set_data方法要快得多：

myobj = imshow(first_image)
for pixel in pixels:
    addpixel(pixel)
    myobj.set_data(segmentedimg)
    draw()

本draw()应确保后端更新图像。

更新： 您的问题已被重大修改。在这种情况下，最好再问一个问题。这是解决第二个问题的方法：

Matplotlib的动画仅处理一个增加的维度（时间），因此您的双循环不会执行。您需要将索引转换为单个索引。这是一个例子：

import numpy as np
from matplotlib import pyplot as plt
from matplotlib import animation

nx = 150
ny = 50

fig = plt.figure()
data = np.zeros((nx, ny))
im = plt.imshow(data, cmap='gist_gray_r', vmin=0, vmax=1)

def init():
    im.set_data(np.zeros((nx, ny)))

def animate(i):
    xi = i // ny
    yi = i % ny
    data[xi, yi] = 1
    im.set_data(data)
    return im

anim = animation.FuncAnimation(fig, animate, init_func=init, frames=nx * ny,
                               interval=50)