Matplotlib 3D散点动画仅绘制首个点问题求助

核心问题分析 #

• 3D散点图更新API误用：set_offsets()是2D散点图的专属方法，3D散点图需要通过_offsets3d属性更新坐标。
• 梯度计算逻辑错误：原代码中梯度的z分量计算完全错误，应该基于当前点的偏导数生成路径，而非固定值代入函数。
• 动画逻辑不符合需求：原代码仅更新单个点的位置，无法实现“保留历史点、依次绘制”的效果。

修复后的完整代码 #

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

def func(x, y):
    return np.cos(x)*np.sin(y)

# 计算函数在(x,y)点的梯度（偏导数）
def gradient_func(x, y):
    df_dx = -np.sin(x)*np.sin(y)  # 对x的偏导
    df_dy = np.cos(x)*np.cos(y)    # 对y的偏导
    return df_dx, df_dy

fig = plt.figure(figsize=(4,4))
fig.tight_layout()

# 生成曲面数据
x = np.linspace(0, 2*np.pi, 100)
y = np.linspace(0, np.pi, 100)
X, Y = np.meshgrid(x, y)
Z = func(X, Y)

# 创建3D轴并绘制曲面
ax = Axes3D(fig, computed_zorder=False)
ax.plot_surface(X, Y, Z, cmap="coolwarm", alpha=0.6)

# 设置坐标轴刻度
x_ticks = [0, np.pi/2, np.pi, 3*np.pi/2, 2*np.pi]
x_labels = ["0", "π/2", "π", "3π/2", "2π"]
y_ticks = [0, np.pi/2, np.pi]
y_labels = ["0", "π/2", "π"]
z_ticks = [-1, 0, 1]
z_labels = ["-1", "0", "1"]
ax.set_xticks(x_ticks, x_labels)
ax.set_yticks(y_ticks, y_labels)
ax.set_zticks(z_ticks, z_labels)
ax.set_title("cos(x)*sin(y)")
ax.grid(False)

# 生成梯度路径点
paso = 0.2  # 调小步长避免路径超出定义域
x_Coord = [0.4*np.pi]
y_Coord = [0.2*np.pi]
z_Coord = [func(x_Coord[0], y_Coord[0])]

# 迭代生成路径点
for n in range(5):
    df_dx, df_dy = gradient_func(x_Coord[n], y_Coord[n])
    new_x = x_Coord[n] + paso * df_dx
    new_y = y_Coord[n] + paso * df_dy
    new_z = func(new_x, new_y)
    x_Coord.append(new_x)
    y_Coord.append(new_y)
    z_Coord.append(new_z)

# 初始化散点图（仅显示第一个点）
scat_1 = ax.scatter([x_Coord[0]], [y_Coord[0]], [z_Coord[0]], c="black", s=50)

def animate(i):
    # 更新散点图，显示前i+1个路径点
    scat_1._offsets3d = (x_Coord[:i+1], y_Coord[:i+1], z_Coord[:i+1])
    return scat_1

# 创建并保存动画
ani = animation.FuncAnimation(fig, animate, frames=len(z_Coord), interval=500)
writer = animation.PillowWriter(fps=1)
ani.save("Imagenes/gradiente1.gif", writer=writer)
plt.show()

关键修复说明 #

1. 替换3D散点更新方法：使用_offsets3d属性传入三个维度的坐标列表，这是Matplotlib 3D散点图的标准更新方式。
2. 修正梯度计算：新增gradient_func计算函数在当前点的偏导数，确保路径严格沿梯度方向生成。
3. 调整动画逻辑：每次帧更新时传入前i+1个点的坐标，实现“逐步绘制所有历史点”的效果。
4. 优化参数：调小步长避免路径超出函数定义域，增加迭代次数生成更多路径点，同时清理了z轴冗余刻度。

内容的提问来源于stack exchange，提问作者Isaibb