python三维图图例

要添加三维图图例，可以使用mpl_toolkits.mplot3d.axes3d.Axes3D对象的legend()方法。以下是一个简单示例：

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

# 生成数据
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))

# 创建图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制三维图形
surf = ax.plot_surface(X, Y, Z, cmap='coolwarm')

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

plt.show()

在这个例子中，我们生成了一个二维网格，并根据一个简单的函数计算了每个点的高度。然后，我们使用`plot_surface()`方法绘制了一个三维表面图，并使用`colorbar()`方法添加了一个图例。请注意，`colorbar()`方法需要传递一个绘制的对象和一些其他参数，例如缩放和方面比。