使用 Matplotlib 创建动画散点图

简介

本实验旨在教你如何使用 Python 的 Matplotlib 库创建动画散点图。我们将涵盖从设置图表到将动画保存为 GIF 的所有内容。在本实验结束时，你将拥有一个可用的动画散点图，可用于可视化你的数据。

虚拟机使用提示

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

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

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

设置图表

创建动画散点图的第一步是设置图表本身。这包括导入必要的库并创建一个图形和轴对象。

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()

定义数据

接下来，我们需要定义将用于散点图的数据。在这个例子中，我们将使用一个从 0 到 10 的简单数字数组。

x = np.linspace(0, 10)

创建散点图

既然我们已经有了数据，就可以创建散点图了。我们通过在轴对象上调用 scatter 函数并传入 x 数据来实现这一点。

scat = ax.scatter(1, 0)

创建动画

最后一步是创建动画。我们使用 animation 模块中的 FuncAnimation 函数来完成此操作。该函数接受几个参数，包括图形对象、将更新绘图的函数以及要使用的帧数。

def animate(i):
    scat.set_offsets((x[i], 0))
    return scat,

ani = animation.FuncAnimation(fig, animate, repeat=True,
                                    frames=len(x) - 1, interval=50)

显示图表

现在，我们可以通过调用 pyplot 模块中的 show 函数来显示图表。

plt.show()

总结

在本实验中，我们学习了如何使用 Python 的 Matplotlib 库创建动画散点图。我们涵盖了从设置图表到将动画保存为 GIF 的所有内容。有了这些知识，现在你可以创建自己的动画散点图来可视化你的数据了。