简单的 Matplotlib 动画教程

Beginner

This tutorial is from open-source community. Access the source code

简介

本教程将指导你如何使用 matplotlib.pyplot 创建一个简单的动画。动画对于可视化随时间变化的数据很有用。在本教程中,我们将生成一组随机数据并将其显示为动画。

虚拟机使用提示

虚拟机启动完成后,点击左上角切换到 笔记本 标签页,以访问 Jupyter Notebook 进行练习。

有时,你可能需要等待几秒钟让 Jupyter Notebook 完成加载。由于 Jupyter Notebook 的限制,操作验证无法自动化。

如果你在学习过程中遇到问题,请随时向 Labby 提问。课程结束后提供反馈,我们将立即为你解决问题。

导入必要的库

我们需要导入必要的库来生成动画。我们将使用 numpy 生成随机数据,并使用 matplotlib.pyplot 将其显示为动画。

import matplotlib.pyplot as plt
import numpy as np

生成随机数据

我们将使用 numpy.random.random() 生成一个三维随机数据数组。我们将使用一个种子值来确保每次运行代码时生成相同的数据集。

np.random.seed(19680801)
data = np.random.random((50, 50, 50))

创建动画

我们将使用一个 for 循环来遍历动画的每一帧。在每次迭代中,我们将清除坐标轴,绘制当前帧,设置标题,并暂停一小段时间以显示动画。

fig, ax = plt.subplots()

for i, img in enumerate(data):
    ax.clear()
    ax.imshow(img)
    ax.set_title(f"frame {i}")
    plt.pause(0.1)

显示动画

我们可以通过运行代码来显示动画。动画将在一个新窗口中显示。

plt.show()

总结

在本教程中,我们学习了如何使用 matplotlib.pyplot 创建一个简单的动画。我们生成了一组随机数据,并使用 for 循环和 plt.pause() 函数将其显示为动画。动画对于可视化随时间变化的数据可能是一个有用的工具。