python三维画图

#绘制三维图形
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
plt.rcParams['font.sans-serif']=['Microsoft YaHei']

fig = plt.figure(figsize=(12,6))
ax = Axes3D(fig)
ax = fig.add_subplot(111, projection='3d')  
#生成代表X轴/Y轴数据的列表，并进行网格化
X,Y=np.mgrid[-3:3:100j,-3:3:100j]
Z=np.exp(-X**2 - Y**2)
#绘制三维图形
surf = ax.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap=plt.get_cmap('rainbow'))
ax.set_zlim(-0.1,1)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
plt.title('$z=exp(-x^2/2-y^2/2)$的三维图形')
fig.colorbar(surf,shrink=0.5,aspect=5)
plt.savefig('../R&Q_pic/test8',dpi=600)
plt.show()