matploylib之热力图

刚学我也不熟，做个笔记吧

 # 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()

效果图：