Python绘制图形

import numpy as np
import matplotlib.pyplot as plt  #导入库
from mpl_toolkits.mplot3d import Axes3D
fig=plt.figure(figsize=(12,6))
ax=Axes3D(fig)
X,Y=np.mgrid[-3:3:100j,-3:3:100j]  #生成X轴/Y轴的数据列表，并进行格网化
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)      #设置z轴的范围
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('test3',dpi=600)
plt.show()