python—使用matplotlib库绘制彩色三维图（曲面绘制）

最近学习过程中需要绘制曲面图，从网上学习了一下，绘制二元函数图，Z= - X ^ 2 - Y ^ 2,程序如下：

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

plt.figure()   
ax = plt.axes(projection="3d")
ax.patch.set_facecolor("white")   #设置 axes 背景颜色
x = numpy.arange(-4,4,0.1)    #0.1为曲面网格x轴长度
y = numpy.arange(-4,4,0.1)    #0.1为曲面网格y轴长度
X,Y = numpy.meshgrid(x,y)    # 生成网格
Z=-X**2-Y**2
ax.plot_surface(X,Y,Z,alpha=0.9,cmap="spring") #生成曲面z，alpha调节透明度，cmap调节曲面色调
#设置X、Y、Z 坐标标签及范围
ax.set_xlabel("X")  
ax.set_xlim(-4,4)  
ax.set_ylabel("Y")
ax.set_ylim(-4,4)
ax.set_zlabel("Z")
plt.show()

程序运行结果：

调整函数z，绘制不同的曲面，如：z=x ^ 2-y ^2 ，曲面颜色：cmap=“winter”

曲面方程：z=3x+2y^2 曲面颜色：cmap=“summer”

曲面方程：z=sqrt((X+1)**2+(Y+5)**2+5) 曲面颜色 ：cmap=“autumn”