matplotlib 中含有 animation 动画模块,使用该模块可以实现一些基本的动画效果。Animation 类是一个动画基类,它针对不同的行为分别派生了不同的子类,主要包括 FuncAnimation 类和 Artist Animation 类。
FuncAnimation 是基于函数的动画类,它通过重复调用同一函数来制作动画。FuncAnimation 类的构造方法的语法格式如下所示:
FuncAnimation(fig, func, frames=None, init_func=None, fargs=None,
save_count=None, *, cache_frame_data=True, **kwargs)
该方法常用参数的含义如下:
- fig:表示动画所在的画布。
- func :表示每帧动画调用的函数。
- frames : 表示动画的长度(一次动画包含的帧数)。
- init_func : 表示用于开始绘制帧的函数,它会会在第一帧动画之前调用一次。若未设置该参数,则程序将使用frames序列中第一项的绘图结结果。
- fargs : 表示传递给 func 函数的其他参数。
- interval : 表示更新动画的频率,以毫秒为单位,默认为200。
- blit : 表示是否更新所有的点,默认为False。官方推荐将 blit 参数设为True,但建议 macOS的用户将blit参数设置为False,否则将无法显示动画。
实例:
绘制一个具有动画效果的图表,要求如下:
(1)绘制一条正弦曲线; (2)绘制一个红色圆点,该圆点最初位于正弦曲线的左端; (3)制作一个圆点沿曲线运动的动画,并时刻显示圆点的坐标位置。
(具体效果参考运行结果gif图)
代码如下:
#导入模版
%matplotlib notebook
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
# 0. 设置中文黑体
plt.rcParams["font.sans-serif"] = ["SimHei"]
plt.rcParams["axes.unicode_minus"] = False
# 1. 准备正弦曲线数据
x = np.linspace(0, 2*np.pi, 100)
y = np.sin(x)
# 2. 创建画布和坐标系
fig = plt.figure(tight_layout=True)
ax = fig.add_subplot(111)
# 3. 绘制正弦曲线
ax.plot(x, y)
# 4. 动画设计
point, = ax.plot(x[0], y[0], "ro") #设置一个动点(圆点)在曲线的最左端
test = ax.text(4, 0.8, "", fontsize=16) #添加无指向型注释文本(在(4,0.8)的位置上显示动点)
def animate(i):
point.set_data(x[i], y[i]) #更新动点位置
test.set_text(f"x={x[i]:.3f}, y={y[i]:.3f}") #更新文本当前的坐标位置(保留小数点后三位)
return point, test
ani = FuncAnimation(fig=fig,
func=animate, #每帧动画调用的函数
frames=len(x), #动画的长度(一次动画包含的帧数)
interval=200, #更新动画的频率(200毫秒)
blit=True #更新所有的点
)
# 5.将动画保存为gif图片
ani.save("pratice7.3.gif", writer='pillow')
# 6.展示图表
plt.show()
运行代码,效果如下所示: