10Contours等高线图

import matplotlib.pyplot as plt
import numpy as np

def f(x,y):
    #the height function
    return (1 - x/2 + x**5 + y**3)*np.exp(-x**2-y**2)
n=256
x=np.linspace(-3,3,n)
y=np.linspace(-3,3,n)
plt.xticks(())
plt.yticks(())

X,Y = np.meshgrid(x,y)

plt.contourf(X,Y,f(X,Y),8,alpha=0.75,cmap=plt.cm.hot)

C = plt.contour(X,Y,f(X,Y),8,colors='black')

plt.clabel(C,inline=True,fontsize=10)


plt.show()