ax.plot3D添加图例

要在3D图中添加图例，可以使用`legend()`方法。下面是一个示例：

import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(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))

# 绘制曲面图
surf = ax.plot_surface(X, Y, Z, cmap='coolwarm')

# 添加图例
fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

在这个例子中，我们生成了一个二维的网格数据，并使用`sin`函数生成了一个曲面。使用`plot_surface()`方法绘制曲面图，并使用`cmap`参数指定颜色映射。然后，使用`colorbar()`方法添加图例。`shrink`参数指定图例的缩放比例，`aspect`参数指定图例的宽高比。