如何在 matplotlib.animation 中正确启用 ffmpeg?


要启用 **matplotlib.animation 的 ffmpeg**,我们可以采取以下步骤:

  • 设置图形大小并调整子图之间和周围的填充。

  • 设置 **ffmpeg** 目录。

  • 使用 **figure()** 方法创建一个新的图形或激活一个现有的图形。

  • 将 **'ax1'** 添加到图形中作为子图排列的一部分。

  • 根据预先存在的轴绘制分割线。

  • 创建要绘制的随机数据,以图像形式(即在二维规则光栅上)显示数据。

  • 为 **ScalarMappable** 实例 **cb** 创建一个颜色条。

  • 将标题设置为当前帧。

  • 制作一个颜色图列表。

  • 通过重复调用函数 animate 来制作动画。该函数创建新的随机数据,然后使用 **imshow()** 方法将数据显示为图像。

  • 获取基于管道的 **ffmpeg** 写入器的实例。

  • 保存当前动画图形。

示例

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from mpl_toolkits.axes_grid1 import make_axes_locatable

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
plt.rcParams['animation.ffmpeg_path'] = 'ffmpeg'

fig = plt.figure()
ax = fig.add_subplot(111)
div = make_axes_locatable(ax)
cax = div.append_axes('right', '5%', '5%')
data = np.random.rand(5, 5)
im = ax.imshow(data)
cb = fig.colorbar(im, cax=cax)
tx = ax.set_title('Frame 0')

cmap = ["copper", 'RdBu_r', 'Oranges', 'cividis', 'hot', 'plasma']

def animate(i):
   cax.cla()
   data = np.random.rand(5, 5)
   im = ax.imshow(data, cmap=cmap[i%len(cmap)])
   fig.colorbar(im, cax=cax)
   tx.set_text('Frame {0}'.format(i))

ani = animation.FuncAnimation(fig, animate, frames=10)
FFwriter = animation.FFMpegWriter()
ani.save('plot.mp4', writer=FFwriter)

输出

执行代码后,它将创建一个名为“plot.mp4”的 mp4 文件并将其保存在项目目录中。


更新于:2021年8月10日

4K+ 次浏览

启动您的 职业生涯

完成课程后获得认证

开始
广告