Python实践:画个动图玩玩,Python绘制GIF图总结

Python实践:画个动图玩玩,Python绘制GIF图总结


上期博客《 Python实践:Pyplot绘图超简洁核心总结》,分享了Python绘图核心方法的体系化总结。

这期分享Python动图制作,可视化绘制动态图,主要用于分析数据的动态变化规律,非常好使,开干。

具体实现


功能需求

  • 输入:不同时刻的一维数据,也即二维数据
  • 过程:绘制每张图,并存入list
  • 输出:一张gif图

实现思路

  • 得到要绘制的不同帧横纵坐标轴数据
  • 设立好图幅:坐标系、标题、标签等
  • 绘制每张图,并用imageio存入list中
  • 绘制完毕,将list中的每帧img,输出成gif

Python代码


demo代码如下:

import matplotlib
matplotlib.use('Agg') # 只绘图保存,不显示图片,一定要加在import后面
import matplotlib.pyplot as plt
import numpy as np
import imageio


def generate_gif_1(data, output_path):
    image_list = []
    ndata = np.array(data)
    row, col = ndata.shape
    x = range(col)
    for i in range(row):
        # plot curves
        # plt.figure(i)
        plt.clf()  # 清除上一幅图,如果不清,则图像不断叠加 clear figure
        plt.cla()  # 清除坐标轴 clear axis
        plt.xlim(0, 12)   # x_min, x_max
        plt.ylim(0, 1.5)  # y_min, y_max
        plt.title('Val Distribu')
        plt.xlabel('Time')
        plt.ylabel('Val')
        plt.grid(linestyle='-.')

        # create gif
        y = ndata[i]
        plt.plot(x[:], y[:], 'b', lw=1)
        plt.savefig('temp.png')
        image_list.append(imageio.imread('temp.png'))
#        plt.show()
        plt.pause(0.1)
        plt.close()

    # save gif
    imageio.mimsave(output_path, image_list, 'GIF', duration=0.35)

if __name__ == "__main__":

    row = 10
    col = 20
    data = np.random.rand(row, col) # 随机生成row行,col列,0-1的数据
    output_path = './demo.gif'

    generate_gif_1(data, output_path)
    
    print('done!')

效果如下:
在这里插入图片描述

参考资料


  1. 【Python可视化】matplotlib画动态曲线,link
  2. 用python做GIF动画,让你的图表动起来, link
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值