ax.plot_surface提取平面图的色调绘制立体图

如果您想在使用 `ax.plot_surface()` 函数绘制三维表面时，使用原始二维图像的色调，则可以使用 `facecolors` 参数。该参数用于设置每个表面的颜色。您可以使用 matplotlib 的 `cm` 模块来选择不同的颜色映射。以下是一个示例代码，演示如何使用 `facecolors` 参数从二维图像中提取色调并绘制三维表面：

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

# 生成二维图像
X = np.linspace(-1.5, 1.5, 100)
Y = np.linspace(-1.5, 1.5, 100)
X, Y = np.meshgrid(X, Y)
Z = np.exp(-X**2 - Y**2)

# 提取色调
norm = plt.Normalize(Z.min(), Z.max())
colors = plt.cm.viridis(norm(Z))

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

# 绘制表面
surf = ax.plot_surface(X, Y, Z, facecolors=colors)

# 添加颜色条
m = plt.cm.ScalarMappable(cmap=plt.cm.viridis)
m.set_array(Z)
plt.colorbar(m)

plt.show()

上述代码将绘制一个二维高斯分布的图像，并使用 `facecolors` 参数从图像中提取色调，然后将这些色调用于绘制三维表面。同时，还添加了一个颜色条，以显示色调对应的值。您可以根据需要选择不同的颜色映射，并使用 `Normalize` 类来设置色调的范围。