ax.plot_surface绘制三维面积图python代码

下面是一个使用ax.plot_surface绘制三维表面图的Python示例代码：

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

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

# 定义 X, Y, Z 三个坐标轴
X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)

# 绘制三维表面图
surf = ax.plot_surface(X, Y, Z, cmap=plt.cm.coolwarm)

# 添加颜色条
fig.colorbar(surf)

# 设置坐标轴标签
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')

# 显示图像
plt.show()

这个示例代码绘制了一个以X、Y坐标为平面，Z坐标为高度的三维表面图。你可以根据自己的需求修改坐标轴范围、数据和绘制效果等参数。