如何使用matplotlib中的plot_surface函数绘制3D图形？

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

# 创建x、y坐标轴上的数据点
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
# 创建网格点坐标矩阵
X, Y = np.meshgrid(x, y)
# 根据X、Y计算Z的值
Z = np.sin(np.sqrt(X**2 + Y**2))

# 创建3D图形对象
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 绘制3D图形
ax.plot_surface(X, Y, Z, cmap='viridis')

# 设置坐标轴标签
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
# 设置图形标题
ax.set_title('3D Surface Plot')

plt.show()