py 画山地图

代码：

import matplotlib.pyplot as plt

import numpy as np
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]
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.jpg',dpi=600)
plt.show()

结果：