本文共 1284 字,大约阅读时间需要 4 分钟。
有时候在展示实验结果的时候,通过一个gif的动态图可以很好进行表现,通过python生成这种gif图的原理实际上就是通过每次生成图像快照,将它们保存下来,然后通过imageio
这个库再将他们组合成一个gif图。
imageio
这个库的话那就先安装: pip install imageio
下面通过模拟一段random walk来生成这个gif图。
import numpy as npimport imageiodef 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)。生成的效果如下:
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)
转载地址:http://awklf.baihongyu.com/