[Matplotlib] 动态视频生成

安装 matplotlib

打开 cmd

pip install matplotlib

ffmpeg 环境搭建

下载压缩包：

1. https://www.gyan.dev/ffmpeg/builds/

2. 访问码：z714

3. csdn

并解压到固定文件夹， 并记录： /Path2Fold/bin/

环境变量注册

在 Path 中添加目录 /Path2Fold/bin/

打开 Python

执行下述代码

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from matplotlib.animation import FFMpegWriter

# 初始化画布
fig, ax = plt.subplots(figsize=(10, 6))
ax.set_xlim(-np.pi/2,  np.pi/2)
ax.set_ylim(-1.5, 1.5)
ax.set_xlabel("x")
ax.set_ylabel("y = cos(t)·cos(x)")
ax.grid(True, alpha=0.3)

# 生成数据
x = np.linspace(-np.pi/2,  np.pi/2, 100)
line, = ax.plot(x, np.cos(x), 'b-', lw=2)

# 更新函数
def update(frame):
    t = frame * 0.1*np.pi
    y = np.cos(t) * np.cos(x)
    line.set_ydata(y)
    ax.set_title(f"Dynamic Wave: t = {t:.2f}")
    return line,

# 创建动画
ani = animation.FuncAnimation(
    fig, update, frames=200, interval=50, blit=True, repeat=False
)

# 保存视频
writer = FFMpegWriter(fps=20, bitrate=5000)
ani.save("cos_wave.mp4", writer=writer)
plt.show()

视频效果

cos_wave