用python画动态图_极简教程: 使用 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()