如何在退出Matplotlib实时绘图程序时保存动态图表至文件？

核心思路 #

避免ani.save()重新运行动画导致挂起，而是在实时绘图过程中缓存所有已生成的帧图像，退出时直接用缓存的帧合成视频，无需重绘或提前固定数据量。

具体实现步骤 #

1. 初始化帧缓存与动画
   定义列表存储每一帧的图像数据，在实时绘图的更新函数中同步捕获并缓存帧：

   import matplotlib.pyplot as plt
   from matplotlib.animation import FuncAnimation
   import numpy as np
   import imageio
   
   # 初始化画布与数据容器
   fig, ax = plt.subplots()
   x_data, y_data = [], []
   line, = ax.plot([], [], 'r-')
   ax.set_xlim(0, 10)
   ax.set_ylim(-1, 1)
   
   # 帧缓存列表
   captured_frames = []
   
   # 实时更新函数：绘图+缓存帧
   def update(frame):
       # 替换为你的实际监控数据获取逻辑
       x = frame
       y = np.sin(x * 0.5)
       x_data.append(x)
       y_data.append(y)
       line.set_data(x_data, y_data)
       
       # 动态调整x轴范围（可选，按需开启）
       if x > ax.get_xlim()[1]:
           ax.set_xlim(x-10, x)
           ax.figure.canvas.draw()
       
       # 捕获当前帧并缓存
       fig.canvas.draw()
       frame_data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
       frame_data = frame_data.reshape(fig.canvas.get_width_height()[::-1] + (3,))
       captured_frames.append(frame_data)
       
       return line,
   
   # 创建动画，关闭默认帧缓存避免冲突
   ani = FuncAnimation(fig, update, interval=200, cache_frame_data=False)
   

2. 绑定退出事件，保存动画
   监听窗口关闭事件，自动将缓存帧合成视频文件：

   def on_close(event):
       # 合成视频，fps需匹配动画interval（interval=200对应fps=5）
       imageio.mimsave('monitor_animation.mp4', captured_frames, fps=5)
       print("动画已保存至monitor_animation.mp4")
   
   # 绑定关闭事件
   fig.canvas.mpl_connect('close_event', on_close)
   
   # 启动实时绘图窗口
   plt.show()
   

关键优势 #

• 无需提前固定数据量：缓存随实时绘图动态增长，直到程序退出
• 避免重复计算：直接使用已绘制的帧图像合成视频，无需重绘数据
• 保留实时查看：原动画逻辑完全保留，仅在后台静默缓存帧

注意事项 #

• 需依赖imageio和imageio-ffmpeg：执行pip install imageio imageio-ffmpeg安装
• 长时运行会占用内存，若仅需保留最近N帧，可在缓存时做截断处理
• 若监控数据更新频率变化，需同步调整fps参数匹配动画interval

内容的提问来源于stack exchange，提问作者CoffeeTableEspresso