Python 如何在matplotlib中使用函数plot_surface绘制三维图？

我的代码如下所示：

将numpy导入为np
将matplotlib.pyplot作为plt导入
从matplotlib导入cm
x_points=np.linspace（1，3，num=3）
y_points=np.linspace（1,6，num=6）
z_points=np.linspace（1，10，num=10）
x_点，y_点，z_点=np.meshgrid（x_点，y_点，z_点）
x_点=x_点。展平（）。重塑（60，3）
y_点=y_点。展平（）。重塑（60，3）
z_点=z_点。展平（）。重塑（60，3）
#为了简化问题并说明我的观点，假设存在某种函数
#x_点、y_点、z_点和u值之间的关系。
u=f（x_点、y_点、z_点）
#规范化的界限：minSpeed和maxSpeed
u=（u-分钟速度）/（最大速度-分钟速度）
图=plt.图（）
ax=图gca（投影=3d'）
im=ax.plot\u曲面（x\u点、y\u点、z\u点、面颜色=cm.coolwarm（u）、线宽=0、抗锯齿=False、阴影=False）

---

我想用u值指示的颜色绘制3D图。我在网上搜索了函数plot_surface，并编写了上面的代码。但它并没有像我预期的那样给出一个情节。如何修改代码以达到我的目的？

plot\U surface

允许显示2个变量的函数Z的变化（例如

Z=np.sqrt（X**2+Y**2）

），但您在这里考虑的是3个变量X、Y和Z的函数U

我建议您通过提供的源代码示例熟悉

plot\u surface

如果您确实试图显示“4D绘图”，请查看