# coding:utf-8

 import numpy as np

 import matplotlib.pyplot as plt

 dx = 0.01

 dy = 0.01

 # x的范围

 x = np.arange(-2.0, 2.0, dx)

 # y的范围

 y = np.arange(-2.0, 2.0, dy)

 # 生成二维网格矩阵

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

 #  定义函数生成三维结构

 def f(x, y):

     return (1 - y ** 5 + x ** 5) * np.exp(-x ** 2 - y ** 2)

 # n条等值线

 n = 8

 #  得到等值线

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

 # 在等值线区域填充颜色

 plt.contourf(X, Y, -f(X, Y), n, cmap=plt.cm.hot)

 # 为等值线注明含义

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

 plt.colorbar()

 plt.show()