使用 matplotlib 创建 3D 动画
要使用 matplotlib 创建 3D 动画,我们可以按以下步骤进行 −
- 导入所需的包。对于 3D 动画,需要从 mpl_toolkits.mplot3d 和 matplotlib.animation 导入 Axes3D。
- 设置图形大小并调整子图之间和周围的填充。
- 使用 numpy 创建 t、x、y 和数据点。
- 创建新图形或激活现有图形。
- 获取 3D 坐标轴的实例。
- 关闭坐标轴。
- 绘制带数据的线。
- 通过重复调用函数 *animate* 来创建动画。
- 要显示图形,请使用 show() 方法。
示例
import numpy as np import matplotlib.pyplot as plt import matplotlib.animation as animation from mpl_toolkits.mplot3d import Axes3D plt.rcParams["figure.figsize"] = [7.00, 3.50] plt.rcParams["figure.autolayout"] = True def animate(num, data, line): colors = ['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728', '#9467bd', '#8c564b', '#e377c2', '#7f7f7f', '#bcbd22', '#17becf'] line.set_color(colors[num % len(colors)]) line.set_alpha(0.7) line.set_data(data[0:2, :num]) line.set_3d_properties(data[2, :num]) return line t = np.arange(0, 20, 0.2) x = np.cos(t) - 1 y = 1 / 2 * (np.cos(2 * t) - 1) data = np.array([x, y, t]) N = len(t) fig = plt.figure() ax = Axes3D(fig) ax.axis('off') line, = plt.plot(data[0], data[1], data[2], lw=7, c='red') line_ani = animation.FuncAnimation(fig, animate, frames=N, fargs=(data, line), interval=50, blit=False) plt.show()
输出
它将产生以下输出
广告