使用 Python 生成影片,无需将单个帧保存到文件
matplotlibpythonserver side programmingprogramming
使用 FuncAnimation 方法,我们可以创建影片。我们将创建一个用户定义的方法 update,以继续更改粒子的位置,最后,该方法将返回散点图实例。
步骤
获取粒子的初始位置、速度、力和大小。
创建一个新图形,或使用 figsize = (7, 7) 激活现有图形。
使用 xlim 和 ylim 向当前图形添加轴并使其成为当前轴。
绘制粒子初始位置的散点图。
通过重复调用函数 *func* 制作动画。我们可以将一个有助于改变粒子位置的用户定义方法传递到 FuncAnimation 类中。
使用 plt.show() 显示图形。
示例
import matplotlib.pyplot as plt from matplotlib.animation import FuncAnimation import numpy as np dt = 0.005 n = 20 L = 1 particles = np.zeros(n, dtype=[("position", float, 2), ("velocity", float, 2), ("force", float, 2), ("size", float, 1)]) particles["position"] = np.random.uniform(0, L, (n, 2)); particles["velocity"] = np.zeros((n, 2)); particles["size"] = 0.5 * np.ones(n); fig = plt.figure(figsize=(7, 7)) ax = plt.axes(xlim=(0, L), ylim=(0, L)) scatter = ax.scatter(particles["position"][:, 0], particles["position"][:, 1]) def update(frame_number): particles["force"] = np.random.uniform(-2, 2., (n, 2)); particles["velocity"] = particles["velocity"] + particles["force"] * dt particles["position"] = particles["position"] + particles["velocity"] * dt particles["position"] = particles["position"] % L scatter.set_offsets(particles["position"]) return scatter, anim = FuncAnimation(fig, update, interval=10) plt.show()