使用matplotlib画简单的等高线图

import matplotlib.pyplot as plt
import numpy as np
#等高线图
x = np.linspace(-5,5,200)
y = np.linspace(-5,5,200)


X, Y = np.meshgrid(x, y)#XY平面的网格数据


Z = (1-X/2+X**7+Y**5)*np.exp(-X**2-Y**2)


#等高线 contour用来画线 contourf用来填充
c = plt.contour(X,Y,Z,8,alpha=0.5,cmap=plt.cm.jet) #10条等高线 cmap = colormap  线的颜色
plt.contourf(X,Y,Z,8,alpha=0.5,cmap=plt.cm.hot) #10条等高线 cmap = colormap  填充颜色


#画线的label
plt.clabel(c,inline=True,fontsize=10,colors='k')


#设置X、Y轴的范围
plt.xlim(-3,3)
plt.ylim(-3,3)


plt.show()