matplotlib绘图8 等高线图

import matplotlib.pyplot as plt
import numpy as np

#计算高度
def f(x,y):
    return (1-x/2+x**5+y**3)*np.exp(-x**2-y**2)

x = np.linspace(-3,3,100)
y = np.linspace(-3,3,100)

#将x,y传入网格中
X,Y = np.meshgrid(x,y)
#设置等高线图及个数 cmap为图的颜色
plt.contourf(X,Y,f(X,Y),8,alpha=0.75,cmap=plt.cm.hot)
#绘制等高线及高度 8代表有8个等高线
C = plt.contour(X,Y,f(X,Y),8,colors='black',linewidths=.5)
#显示等高线的值 在等高线的中间 字体大小为10
plt.clabel(C,inline=True,fontsize=10)
#去掉坐标值
plt.xticks(())
plt.yticks(())

plt.show()