Ctrl+K

注意

转到末尾 下载完整示例代码。

使用 Matplotlib 进行动画#

基于其绘图功能，Matplotlib 还提供了一个使用 animation 模块生成动画的接口。动画是一系列帧，其中每一帧对应于 Figure 上的绘图。本教程涵盖了有关如何创建此类动画以及可用不同选项的一般指南。

import matplotlib.pyplot as plt
import numpy as np

import matplotlib.animation as animation

动画类#

Matplotlib 中的动画过程可以从 2 种不同的方式来考虑

  • FuncAnimation: 生成第一帧的数据，然后修改此数据以创建动画绘图。

  • ArtistAnimation: 生成一个艺术家列表（可迭代的），这些艺术家将在动画中的每一帧中绘制。

FuncAnimation 在速度和内存方面更有效，因为它只绘制一次艺术家，然后对其进行修改。另一方面，ArtistAnimation 很灵活，因为它允许任何艺术家可迭代对象按顺序进行动画处理。

FuncAnimation#

FuncAnimation 类允许我们通过传递一个迭代修改绘图数据的函数来创建动画。 这是通过使用各种 Artist （例如：Line2D，PathCollection 等）的 *setter* 方法来实现的。 一个典型的 FuncAnimation 对象接受一个我们想要动画化的 Figure 和一个修改绘图上数据的函数 *func*。 它使用 *frames* 参数来确定动画的长度。 *interval* 参数用于确定两帧之间绘制的时间（以毫秒为单位）。 使用 FuncAnimation 进行动画制作通常需要以下步骤

  1. 像在静态绘图中一样绘制初始图形。 将所有创建的艺术家（由绘图函数返回）保存在变量中，以便您以后在动画函数中访问和修改它们。

  2. 创建一个动画函数，该函数更新给定帧的艺术家。 通常，这会调用艺术家的 set_* 方法。

  3. 创建一个 FuncAnimation，传递 Figure 和动画函数。

  4. 使用以下方法之一保存或显示动画

下表显示了一些绘图方法、它们返回的图形以及一些常用的 set_* 方法，这些方法更新底层数据。虽然更新数据是动画中最常见的操作，但您也可以更新其他方面，例如颜色或文本位置。

绘图方法

图形

数据设置方法

Axes.plot

lines.Line2D

set_data, set_xdata, set_ydata

Axes.scatter

collections.PathCollection

set_offsets

Axes.imshow

image.AxesImage

AxesImage.set_data

Axes.annotate

text.Annotation

update_positions

Axes.barh

patches.Rectangle

set_angle, set_bounds, set_height, set_width, set_x, set_y, set_xy

Axes.fill

patches.Polygon

set_xy

Axes.add_patch(patches.Ellipse)

patches.Ellipse

set_angle, set_center, set_height, set_width

Axes.set_title, Axes.text

text.Text

set_text

本教程无法涵盖所有类型艺术家的设置方法，但可以在各自的文档中找到。以下是如何在 Axes.scatter 和 Axes.plot 中使用这些更新方法的示例。

fig, ax = plt.subplots()
t = np.linspace(0, 3, 40)
g = -9.81
v0 = 12
z = g * t**2 / 2 + v0 * t

v02 = 5
z2 = g * t**2 / 2 + v02 * t

scat = ax.scatter(t[0], z[0], c="b", s=5, label=f'v0 = {v0} m/s')
line2 = ax.plot(t[0], z2[0], label=f'v0 = {v02} m/s')[0]
ax.set(xlim=[0, 3], ylim=[-4, 10], xlabel='Time [s]', ylabel='Z [m]')
ax.legend()


def update(frame):
    # for each frame, update the data stored on each artist.
    x = t[:frame]
    y = z[:frame]
    # update the scatter plot:
    data = np.stack([x, y]).T
    scat.set_offsets(data)
    # update the line plot:
    line2.set_xdata(t[:frame])
    line2.set_ydata(z2[:frame])
    return (scat, line2)


ani = animation.FuncAnimation(fig=fig, func=update, frames=40, interval=30)
plt.show()

ArtistAnimation#

ArtistAnimation 可用于生成动画，前提是数据存储在不同的艺术家对象上。然后，这些艺术家对象的列表逐帧转换为动画。例如，当我们使用 Axes.barh 绘制条形图时，它会为每个条形和误差线创建多个艺术家对象。要更新绘图，需要单独更新容器中的每个条形并重新绘制它们。相反，animation.ArtistAnimation 可用于单独绘制每一帧，然后将它们拼接在一起形成动画。条形图竞赛就是一个简单的例子。

fig, ax = plt.subplots()
rng = np.random.default_rng(19680801)
data = np.array([20, 20, 20, 20])
x = np.array([1, 2, 3, 4])

artists = []
colors = ['tab:blue', 'tab:red', 'tab:green', 'tab:purple']
for i in range(20):
    data += rng.integers(low=0, high=10, size=data.shape)
    container = ax.barh(x, data, color=colors)
    artists.append(container)


ani = animation.ArtistAnimation(fig=fig, artists=artists, interval=400)
plt.show()

动画写入器#

动画对象可以使用各种多媒体写入器（例如：Pillow、ffpmeg、imagemagick）保存到磁盘。并非所有视频格式都受所有写入器支持。主要有四种类型的写入器。

  • PillowWriter - 使用 Pillow 库创建动画。

  • HTMLWriter - 用于创建基于 JavaScript 的动画。

  • 基于管道的写入器 - FFMpegWriter 和 ImageMagickWriter 是基于管道的写入器。这些写入器将每一帧管道传输到实用程序（ffmpeg / imagemagick），然后将所有帧拼接在一起以创建动画。

  • 基于文件的写入器 - FFMpegFileWriter 和 ImageMagickFileWriter 是基于文件的写入器的示例。这些写入器比基于管道的写入器速度慢，但对于调试更有用，因为它们会在将帧拼接成动画之前将每一帧保存到文件中。

保存动画#

编写器

支持的格式

PillowWriter

.gif, .apng, .webp

HTMLWriter

.htm, .html, .png
FFMpegWriter
FFMpegFileWriter

所有由 ffmpeg 支持的格式：ffmpeg -formats
ImageMagickWriter
ImageMagickFileWriter

所有由 imagemagick 支持的格式：magick -list format

要使用任何编写器保存动画，我们可以使用 animation.Animation.save 方法。它接受我们要保存动画的文件名和编写器，编写器可以是字符串或编写器对象。它还接受一个fps参数。此参数与 FuncAnimation 或 ArtistAnimation 使用的interval参数不同。fps决定保存的动画使用的帧速率，而interval决定显示的动画使用的帧速率。

以下是一些示例，展示了如何使用不同的编写器保存动画。

Pillow 编写器

ani.save(filename="/tmp/pillow_example.gif", writer="pillow")
ani.save(filename="/tmp/pillow_example.apng", writer="pillow")

HTML 编写器

ani.save(filename="/tmp/html_example.html", writer="html")
ani.save(filename="/tmp/html_example.htm", writer="html")
ani.save(filename="/tmp/html_example.png", writer="html")

FFMpegWriter

ani.save(filename="/tmp/ffmpeg_example.mkv", writer="ffmpeg")
ani.save(filename="/tmp/ffmpeg_example.mp4", writer="ffmpeg")
ani.save(filename="/tmp/ffmpeg_example.mjpeg", writer="ffmpeg")

Imagemagick 编写器

ani.save(filename="/tmp/imagemagick_example.gif", writer="imagemagick")
ani.save(filename="/tmp/imagemagick_example.webp", writer="imagemagick")
ani.save(filename="apng:/tmp/imagemagick_example.apng",
         writer="imagemagick", extra_args=["-quality", "100"])

(apng 的 extra_args 用于将文件大小减少约 10 倍)

脚本总运行时间：（0 分钟 7.598 秒）

由 Sphinx-Gallery 生成的画廊

本页