极简教程: 使用 matplotlib 绘制 GIF 动图

开门见山，直接上例子：

有如下特点：

• 散点图的部分是不变的；线是移动的
• X 轴标题每一祯改变一次

DEMO 的环境

• Ubuntu 18.04.2 LTS
• conda 4.6.3
• Python 3.7.2

创建 virtualenv

ichexw at n3xt-Studio -> conda create --name matplot-gif python=3.7
ichexw at n3xt-Studio -> conda activate matplot-gif

安装必要的依赖

安装 matplotlib

(matplotlib-gif) ichexw at n3xt-Studio -> conda install matplotlib

安装 imagemagick

(matplotlib-gif) ichexw at n3xt-Studio -> conda install -c conda-forge imagemagick

代码实现

import sys
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
# 创建图层和布局
fig, ax = plt.subplots()
fig.set_tight_layout(True)
# 查看图标的尺寸。如果你保存成 gif 的时候，你需要提供 DPI
print('fig size: {0} DPI, size in inches {1}'.format(
fig.get_dpi(), fig.get_size_inches()))
# 绘制一个散点图（不会重绘），和初始的线
x = np.arange(0, 20, 0.1)
ax.scatter(x, x + np.random.normal(0, 3.0, len(x)))
line, = ax.plot(x, x - 5, 'r-', linewidth=2)
def update(i):
label = 'timestep {0}'.format(i)
print(label)
# 更新线和坐标轴标签
line.set_ydata(x - 5 + i)
ax.set_xlabel(label)
# 返回要重绘的对象
return line, ax
if __name__ == '__main__':
# FunAnimation 将会在每一帧执行一次 update
# frames: 帧数
# interval: 每帧的间隔
anim = FuncAnimation(fig, update, frames=np.arange(0, 10), interval=200)
if len(sys.argv) > 1 and sys.argv[1] == 'save':
# 如果第一参数是 save，教会保存成 gif
# **重点**
# dpi: 保存的尺寸
# writer: 使用的渲染器，我们制定成 imagemagick
anim.save('line.gif', dpi=80, writer='imagemagick')
else:
# 否则直接展示
plt.show()