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

我想通过在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函数。我得到的结果如下：
我怎样才能避免这种减速呢？

最佳答案

事情变慢了，因为你在添加越来越多的图像，并且每次都在绘制它们。
要么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()

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