如何为分段定义函数的plot_surface实现Matplotlib动画？

核心思路 #

问题根源在于你每次更新时重新创建plot_surface对象，而非复用已有对象更新数据。另外，改用numpy向量化操作替代逐点判断，能让z值的初始化和更新更简洁高效。

完整实现代码 #

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

# 创建x、y网格
x = np.linspace(-30, 30, 100)
y = np.linspace(-30, 30, 100)
X, Y = np.meshgrid(x, y)
r = np.sqrt(X**2 + Y**2)  # 预计算径向距离，避免重复计算

# 初始化figure和3D轴
fig = plt.figure(figsize=(10, 8))
ax = fig.add_subplot(projection='3d')
ax.set_xlim(-30, 30)
ax.set_ylim(-30, 30)
ax.set_zlim(-1, 1)  # 根据函数值域预设z轴范围
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('f(x,y,t)')

# 初始化t=0时的z值
t_init = 0
Z = np.where(r < t_init, np.exp(-t_init/10)*np.sin(r - t_init), 0)
# 创建初始曲面并保存对象
surface = ax.plot_surface(X, Y, Z, cmap='viridis', alpha=0.8)

def update(frame):
    # frame对应当前t值，从0到35逐步变化
    t = frame
    # 用numpy向量化操作计算当前t对应的z值
    new_Z = np.where(r < t, np.exp(-t/10)*np.sin(r - t), 0)
    # 更新已有曲面的z数据
    surface.set_zdata(new_Z)
    # 更新标题显示当前t值
    ax.set_title(f't = {t:.1f}')
    return surface,

# 创建动画，frames设置为0到35的序列，步长0.5可调整动画速度
ani = FuncAnimation(fig, update, frames=np.arange(0, 35, 0.5), blit=True)

plt.show()

关键细节说明 #

• 复用曲面对象：初始化时保存plot_surface返回的Poly3DCollection对象，在update函数中通过set_zdata更新数据，避免重复绘制新曲面，这是解决动画失效的核心。
• 向量化计算：用np.where直接处理整个网格的分段函数，无需逐点循环，代码更简洁高效。
• 预计算径向距离：提前计算r = sqrt(X²+Y²)，避免每次update重复计算，提升动画流畅度。
• Blitting优化：设置blit=True让动画只更新变化的部分，大幅提升流畅度。

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