为什么每次更新时我的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()

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