使用 matplotlib.animation 从 matplotlib 制作视频后创建的空 .mp4 文件

Question

我找到了这个小代码，我可以使用它保存一个视频（在网格中随机改变颜色）：

import numpy as np
import matplotlib
matplotlib.use("Agg")
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from scipy import rand

Writer = animation.writers['ffmpeg']
writer = Writer(fps=15, metadata=dict(artist='xy'), bitrate=3600)

fig = plt.figure()
frames = []
for add in np.arange(15):
    base = rand(10, 10)
    frames.append((plt.pcolormesh(base, ),))

im_ani = animation.ArtistAnimation(fig, frames, interval=500, repeat_delay=3000,
                               blit=True)
im_ani.save('Vid.mp4', writer=writer)

我尝试将其插入到我的模拟中，我没有收到任何错误，但视频是空的，整个画面都是白色的。你能帮我解决这个问题吗？这是我得到它的最简单的情况：

class Dummy():

    def __init__(self):
        self.video=[]

    def addFrame(self):

        Frame=rand(10,10)
        print (Frame)
        self.video.append((plt.pcolormesh(Frame),))

    def saveVideo(self):

        Writer = animation.writers['ffmpeg']
        writer = Writer(fps=15, metadata=dict(artist='XY'), bitrate=3600)
        fig = plt.figure()

        im_ani = animation.ArtistAnimation(fig, self.video, interval=500, repeat_delay=3000,
                               blit=True)
        im_ani.save('myVid.mp4', writer=writer, dpi=dpi)

你可以试试：

from scipy import rand

foo=Dummy()

for i in range(20):
   foo.addFrame()

foo.saveVideo()

Answer 1

您的self.video.append((plt.pcolormesh(Frame),)) 行很好。你只是改变了语句的顺序。

import matplotlib
matplotlib.use("Agg")
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import numpy as np


class Dummy():
    def __init__(self, fname):
        self.fname = fname
        self.video = []
        self.fig, self.ax = plt.subplots()  # Create new figure here

    def addFrame(self):
        Frame = np.random.rand(10, 10)
        self.video.append((self.ax.pcolormesh(Frame), ))

    def saveVideo(self):
        Writer = animation.writers['ffmpeg']
        writer = Writer(fps=15, metadata=dict(artist='XY'), bitrate=3600)
        im_ani = animation.ArtistAnimation(self.fig, self.video, interval=500,
                                           repeat_delay=3000, blit=True)
        im_ani.save(self.fname, writer=writer)


foo1 = Dummy('foo1.mp4')
for i in range(20):
   foo1.addFrame()
foo1.saveVideo()

foo2 = Dummy('foo2.mp4')
for i in range(40):
   foo2.addFrame()
foo2.saveVideo()