Matplotlib 视频创建

编辑：欧内斯特的重要性提供了答案，但是我仍然邀请大家解释，为什么savefig逻辑不同于animation logic.

我想制作视频matplotlib。我浏览了手册和示例，但我就是不明白。 （关于matplotlib，我总是复制示例，因为经过五年的python和两年的mathplotlib，我仍然理解0.0％的matplotlib语法）

六个小时后，这就是我的想法。好吧，我得到的是空视频。不知道为什么。

import os
import math
import matplotlib
matplotlib.use("Agg")
from matplotlib import pyplot as plt
import matplotlib.animation as animation

# Set up formatting for the movie files
Writer = animation.writers['ffmpeg']
writer = Writer(fps=15, metadata=dict(artist='Me'), bitrate=1800)

numb=100
temp=[0.0]*numb
cont=[0.0]*numb
for i in range(int(4*numb/10),int(6*numb/10)):
    temp[i]=2
    cont[i]=2

fig = plt.figure()
plts=fig.add_subplot(1,1,1)

plts.set_ylim([0,2.1])
plts.get_xaxis().set_visible(False)
plts.get_yaxis().set_visible(False)
ims = []

for i in range(1,10):
    line1, = plts.plot(range(0,numb),temp, linewidth=1, color='black')
    line2, = plts.plot(range(0,numb),cont, linewidth=1, color='red')
#    savefig is here for testing, works perfectly!
#    fig.savefig('test'+str(i)+'.png', bbox_inches='tight', dpi=300)
    ims.append([line1,line2])
    plts.lines.remove(line1)
    plts.lines.remove(line2)
    for j in range(1,10):
        tempa=0
        for k in range (1,numb-1):
            tempb=temp[k]+0.51*(temp[k-1]-2*temp[k]+temp[k+1])
            temp[k-1]=tempa
            tempa=tempb
        temp[numb-1]=0
    for j in range(1,20):
        conta=0
        for k in range (1,numb-1):
            contb=cont[k]+0.255*(cont[k-1]-2*cont[k]+cont[k+1])
            cont[k-1]=conta
            conta=contb
        cont[numb-1]=0

im_ani = animation.ArtistAnimation(fig, ims, interval=50, repeat_delay=3000,blit=True)
im_ani.save('im.mp4', writer=writer)

有人可以帮我弄这个吗？

如果你想要一个不为空的图，主要的想法是不要从图中删除线条。

也就是删除这两行

plts.lines.remove(line1)
plts.lines.remove(line2)

如果删除这两行，输出将如下所示

[Link to orginial size animation]

现在有人可能会问，为什么我不需要在每个迭代步骤中删除艺术家，否则所有线条都会立即填充画布？

答案是 ArtistAnimation 负责处理这个问题。它只会显示提供的列表中与给定时间步相对应的艺术家。因此，在 for 循环结束时，您最终会将所有线条绘制到画布上，但一旦动画开始，它们将全部被删除，并且一次仅显示一组艺术家。

在这种情况下，使用循环来保存各个图像当然不是一个好主意，因为最终图像将同时包含所有绘制的线条，

然后，解决方案是运行两次脚本，一次用于动画，另一次在每个时间步中删除线条。或者，也许更好，使用动画本身来创建图像。

im_ani.save('im.png', writer="imagemagick") 

将创建图像为im-<nr>.png在当前文件夹中。需要安装 imagemagick。

1.我在删除 line1 和 line2 之前附加了它们。但他们在最终结果中仍然消失了。怎么会？