python中绘制三维曲面图

在Python中，可以使用matplotlib库来绘制三维曲面图。首先，需要导入必要的库和模块。然后，可以使用plt.figure()生成一个画布，并使用plt.axes(projection='3d')生成一个三维坐标轴。接下来，可以使用numpy库生成数据，并使用ax.plot_surface()函数绘制曲面图。最后，使用plt.show()显示图形。

下面是一个示例代码：

import numpy as np
from matplotlib import pyplot as plt

fig = plt.figure()
ax = plt.axes(projection="3d")

x = y = np.arange(start=-4, stop=4, step=0.1)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2

ax.plot_surface(X, Y, Z, alpha=0.9, cmap='rainbow')

plt.show()

在这个示例中，我们使用了numpy库生成了x和y的坐标点，并使用meshgrid函数将它们组合成网格。然后，根据函数Z = X**2 + Y**2生成了z的坐标点。最后，使用ax.plot_surface()函数绘制了曲面图，并使用cmap参数指定了颜色映射为'rainbow'。