如何用Python将3D轨迹绘图生成并保存为GIF？

我有一段Python代码，能生成由x、y、z定义的轨迹的多幅3D绘图。现在只能把所有绘图存到本地硬盘，再用第三方软件转成GIF，想问下怎么直接用Python生成并保存这个GIF？

原代码 #

for t in range(len(theta)):
    fig = plt.figure('Parametrinai blynai')
    ax = fig.add_subplot(111, projection='3d')
    ax.plot(x[1:t], y[1:t], z[1:t], '-r', linewidth=3)

    ax.set_xlabel('X', fontweight='bold', fontsize=14)
    ax.set_ylabel('Y', fontweight='bold', fontsize=14)
    ax.set_zlabel('Z', fontweight='bold', fontsize=14)
    ax.set_xlim([-20, 20])
    ax.set_ylim([-20, 20])
    ax.set_zlim([-10, 10])
    plt.title('Parametrinis blynas', fontweight='bold', fontsize=16)

    ax.quiver(x[t - 1], y[t - 1], z[t - 1], emp * T[t, 0], emp * T[t, 1], emp * T[t, 2], color='k')
    ax.quiver(x[t - 1], y[t - 1], z[t - 1], emp * B[t, 0], emp * B[t, 1], emp * B[t, 2], color='b')
    ax.quiver(x[t - 1], y[t - 1], z[t - 1], emp * N[t, 0], emp * N[t, 1], emp * N[t, 2], color='g')
    pavadinimas = str(t)
    plt.savefig(pavadinimas, dpi=300)

方法一：用imageio自动合成帧 #

要是不想大改原代码，直接用imageio库把已经保存的图片合成GIF就行，不用手动开第三方软件。步骤如下：

1. 先安装imageio：

pip install imageio

2. 原代码跑完之后，加这段代码读取所有图片生成GIF：

import imageio
import os

frames = []
for t in range(len(theta)):
    img_path = f"{t}.png"
    frames.append(imageio.imread(img_path))
    # 可以顺便删掉临时图片，节省空间
    os.remove(img_path)

# duration是每帧停留的时间（秒），按需调整
imageio.mimsave('trajectory.gif', frames, duration=0.1)

方法二：用matplotlib动画模块直接生成 #

更高效的方式是用matplotlib的FuncAnimation直接制作动画，不用保存中间帧。修改后的代码如下：

import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

# 先初始化画布和坐标轴，只做一次就行，不用循环创建
fig = plt.figure('Parametrinai blynai')
ax = fig.add_subplot(111, projection='3d')
ax.set_xlabel('X', fontweight='bold', fontsize=14)
ax.set_ylabel('Y', fontweight='bold', fontsize=14)
ax.set_zlabel('Z', fontweight='bold', fontsize=14)
ax.set_xlim([-20, 20])
ax.set_ylim([-20, 20])
ax.set_zlim([-10, 10])
plt.title('Parametrinis blynas', fontweight='bold', fontsize=16)

# 初始化轨迹线和三个箭头，一开始是空的
trajectory, = ax.plot([], [], [], '-r', linewidth=3)
quiver_T = ax.quiver([], [], [], [], [], [], color='k')
quiver_B = ax.quiver([], [], [], [], [], [], color='b')
quiver_N = ax.quiver([], [], [], [], [], [], color='g')

# 每帧更新的函数
def update(t):
    # 更新轨迹数据
    trajectory.set_data(x[1:t], y[1:t])
    trajectory.set_3d_properties(z[1:t])
    
    # 更新箭头：先删除旧的，再绘制新的
    global quiver_T, quiver_B, quiver_N
    quiver_T.remove()
    quiver_B.remove()
    quiver_N.remove()
    quiver_T = ax.quiver(x[t-1], y[t-1], z[t-1], emp*T[t,0], emp*T[t,1], emp*T[t,2], color='k')
    quiver_B = ax.quiver(x[t-1], y[t-1], z[t-1], emp*B[t,0], emp*B[t,1], emp*B[t,2], color='b')
    quiver_N = ax.quiver(x[t-1], y[t-1], z[t-1], emp*N[t,0], emp*N[t,1], emp*N[t,2], color='g')
    
    return trajectory, quiver_T, quiver_B, quiver_N

# 创建动画：frames是要渲染的帧数，interval是每帧间隔（毫秒）
ani = FuncAnimation(fig, update, frames=range(2, len(theta)), interval=100, blit=True)

# 保存GIF，需要先安装pillow：pip install pillow
ani.save('trajectory.gif', writer='pillow', dpi=300)
plt.show()

注意：

• frames从2开始，因为t=1时x[1:t]是空的，会触发报错；
• blit=True能让动画更流畅，只更新画面中变化的部分；
• 要是觉得GIF速度不合适，调整interval（毫秒）或者duration（秒）参数即可。

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