【python】如何用python生成并保存gif动态图

有时候在展示实验结果的时候，通过一个gif的动态图可以很好进行表现，通过python生成这种gif图的原理实际上就是通过每次生成图像快照，将它们保存下来，然后通过imageio这个库再将他们组合成一个gif图。
如果没有安装imageio这个库的话那就先安装：

pip install imageio

下面通过模拟一段random walk来生成这个gif图。

import numpy as np
import imageio

def random_walk():
    fig = plt.figure(figsize=(16,9))
    ax = fig.add_subplot(111)
    data = np.random.normal(size=50)
    sum = 0
    sum_list = []
    image_list = []
    for i in range(len(data)):
        sum += data[i]
        sum_list.append(sum)
        ax.plot(range(len(sum_list)), sum_list)
        plt.savefig('temp.png')
        image_list.append(imageio.imread('temp.png'))
    imageio.mimsave('pic.gif', image_list, duration=1)

代码很简单，只需把每次生成的图像保存下来，然后放到一个list里面，最后通过imageio.mimsave()函数进行保存即可，其中参数duration是生成的gif图中每一张图像之间的时间间隔（s）。生成的效果如下：
因为上面的代码是在同一个fig中进行绘制的，所以每次生成的曲线颜色会不一样，如果要保持一致，只需要每次重新创建一个fig即可，代码如下：

def random_walk2():
    data = np.random.normal(size=40)
    sum = 0
    sum_list = []
    image_list = []
    for i in range(len(data)):
        fig = plt.figure(figsize=(16, 9))
        ax = fig.add_subplot(111)
        sum += data[i]
        sum_list.append(sum)
        ax.plot(range(len(sum_list)), sum_list)
        plt.savefig('temp.png')
        plt.close()
        image_list.append(imageio.imread('temp.png'))
    imageio.mimsave('pic.gif', image_list, duration=1)