使用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()
输出
它将生成以下输出