在Matplotlib动画中更新surface_plot上的z数据

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

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

x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))
surface_plot = ax.plot_surface(X, Y, Z, cmap='viridis')

def update(frame):
    # 计算新的z数据
    new_Z = np.sin(np.sqrt(X**2 + Y**2 + frame/10))
    # 更新surface_plot的z数据
    surface_plot.set_array(new_Z.ravel())
    return surface_plot,

ani = animation.FuncAnimation(fig, update, frames=100, interval=50)

plt.show()