python - 为什么我的 pylab 动画每次更新都会变慢？

Question

我想通过在 for 循环中调用 imshow 来显示一个简单的动画。这是我的问题的演示:

import pylab,time
images = [pylab.uniform(0,255,(50,50)) for _ in xrange(40)]
pylab.ion()
timings = []
for img in images:
  tic = time.time()
  pylab.imshow(img)
  pylab.draw()
  toc = time.time()
  timings.append(toc-tic)
pylab.clf()
pylab.plot(timings)
pylab.title('elapsed time per iteration')
pylab.ioff()
pylab.show()

请注意，我在运行循环之前生成图像，并且我计时的唯一部分是 imshow 和 draw 函数。我得到的结果如下所示:

我怎样才能避免这种减速？

Answer 1

事情正在变慢，因为您要添加越来越多的图像并每次绘制它们所有。

要么 1) 清除每张图像之间的图(在您的情况下，pylab.cla())，要么更好 2) 不要制作新图像，只需设置现有图像到新数据。

作为使用 cla() 的例子:

import matplotlib.pyplot as plt
import numpy as np

images = np.random.uniform(0, 255, size=(40, 50, 50))

fig, ax = plt.subplots()

fig.show()
for image in images:
    ax.imshow(image)
    fig.canvas.draw()
    ax.cla()

作为仅设置数据的示例:

import matplotlib.pyplot as plt
import numpy as np

images = np.random.uniform(0, 255, size=(40, 50, 50))

fig, ax = plt.subplots()

im = ax.imshow(images[0])
fig.show()
for image in images[1:]:
    im.set_data(image)
    fig.canvas.draw()

您会注意到第二种方法要快得多。