标签:Python,Matplotlib库
Matplotlib是一个强大的Python绘图库,但许多人可能不知道它能够创建动画图。
图1
什么是动画?
简单地说,动画由一系列略微有些变化的静态图像组成。当把这些静态图像放在一起并快速翻阅时,我们的眼睛和大脑会欺骗我们,让我们觉得这些图像在移动(尽管它们不是)。
基于这种理解,要制作动画绘图,需要:
1.创建大量静态图像/图表
2.把它们按顺序翻动
库
可以使用pip安装matplotlib,只需在命令提示符/终端窗口中键入以下内容:
pip install matplotlib
使用Matplotlib制作动画
将用matplotlib制作一个简单的动画:一个在半径为1.5的圆中移动的红点。
首先,需要一个函数来计算圆的x和y坐标。这里没有使用x^2+y^2=r^2公式,而是使用圆的参数表示:
x = cos(t)
y = sin(t)
其中t是一个角度量
%matplotlib notebook
import matplotlib.pyplot as plt
import math
from matplotlib.animation import FuncAnimation
def circle_coords(r, step):
coords = []
t = 0
while t < 2 * math.pi:
coords.append((r*math.cos(t),r*math.sin(t)))
t += step
return coords
coords = circle_coords(1.5, 0.1)
代码运行后,coords内的结果如下图2所示。
图2
步骤1:创建一个静态图表
为坐标系coords中的第一组(x,y)坐标创建一个带有红点的图表。
fig, ax = plt.subplots()
ax.scatter(x=coords[0][0],y=coords[0][1],c='red', marker = 'o')
此代码在(1.5,0)处创建一个红点:
图3
步骤2:创建许多静态图表
下一步是创建大量略有变化的静态图表。将使用前5个坐标创建5个图表来说明这一点:
fig, axs = plt.subplots(nrows = 1,ncols = 5, figsize = (10, 5), tight_layout = True)
for i, ax in enumerate(axs):
ax.scatter(x=coords[i][0],y=coords[i][1], c='red', marker='o')
ax.set_xlim([0,2])
ax.set_ylim([-0.5,2])
结果如下图4所示。
图4
如此狭窄的图表上是想说明,当浏览每个图表时,圆点是向上和向右移动的。
Matplotlib中的FuncAnimation模块
从技术上讲,只需要创建多个静态图表(每个坐标对应一个),然后将它们组合在一起,就会有一个动画。然而,matplotlib库提供了一个FuncAnimation模块来实现这一点。
要使用FuncAnimation,需要一个函数来绘制静态图表。绘图函数的名称无关紧要,这里将其命名为update():
1.绘图函数接受1个参数i,这意味着帧编号。
2.该函数将在每次调用开始时清除图表,因此在每一帧只会看到一个红点。
3.ax.set_xlim()和ax.set_ylim()函数以确保修复x轴和y轴。
4.ax.scatter()一次只画一个红点。
from matplotlib.animation import FuncAnimation
def update(i):
ax.clear()
ax.set_facecolor(plt.cm.Blues(.2))
ax.set_xlim([-2,2])
ax.set_ylim([-2,2])
ax.set_title('circling')
ax.scatter(x=coords[i][0],y=coords[i][1],c='red',marker='o')
[spine.set_visible(False) for spine in ax.spines.values()] #移除图表轮廓
现在,有了绘图函数,可以创建FuncAnimation对象,它接受以下参数:
1.fig:要绘制的图形对象
2.func:绘图/更新函数
3.frames:要创建的帧数(静态图像)
4.interval:帧间延迟(毫秒)
甚至可以将动画保存为gif文件并发送给其他人:
fig, ax =plt.subplots(figsize=(6,6))
anime = FuncAnimation(
fig = fig,
func = update,
frames = len(coords),
interval = 50
)
anime.save('circle.gif')
有兴趣的朋友可以试试看!