In [6]:

    

import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline

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)
X, Y = np.meshgrid(x, y)

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

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

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


    

    
Out[6]:





([], <a list of 0 Text yticklabel objects>)