python动态三维绘图代码怎么写

45次阅读

共计 1249 个字符，预计需要花费 4 分钟才能阅读完成。

在 Python 中，可以使用以下方法进行动态三维绘图：

导入必要的库：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib.animation import FuncAnimation

创建一个空的三维图形对象：

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

创建一个空的线条对象：

line, = ax.plot([], [], [], 'b-', lw=2)

定义初始化函数，用于初始化图形对象的状态：

def init():
    line.set_data([], [])
    line.set_3d_properties([])
    return line,

定义更新函数，用于更新图形对象的状态：

def update(frame):
    # 根据帧数 frame 计算新的数据点 
    x = np.linspace(0, 2 * np.pi, 100)
    y = np.sin(x + frame * 0.1)
    z = np.cos(x + frame * 0.1)

    # 更新线条对象的数据 
    line.set_data(x, y)
    line.set_3d_properties(z)
    return line,

创建动画对象，并设置参数：

ani = FuncAnimation(fig, update, frames=100, init_func=init, blit=True)

显示动画：

plt.show()

完整的代码示例：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib.animation import FuncAnimation

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
line, = ax.plot([], [], [], 'b-', lw=2)

def init():
    line.set_data([], [])
    line.set_3d_properties([])
    return line,

def update(frame):
    x = np.linspace(0, 2 * np.pi, 100)
    y = np.sin(x + frame * 0.1)
    z = np.cos(x + frame * 0.1)

    line.set_data(x, y)
    line.set_3d_properties(z)
    return line,

ani = FuncAnimation(fig, update, frames=100, init_func=init, blit=True)
plt.show()