matploylib之热力图
刚学我也不熟，做个笔记吧
1 # coding:utf-8 2 import numpy as np 3 import matplotlib.pyplot as plt 4 5 dx = 0.01 6 dy = 0.01 7 # x的范围 8 x = np.arange(-2.0, 2.0, dx) 9 # y的范围 10 y = np.arange(-2.0, 2.0, dy) 11 # 生成二维网格矩阵 12 X, Y = np.meshgrid(x, y) 13 14 15 # 定义函数生成三维结构 16 def f(x, y): 17 return (1 - y ** 5 + x ** 5) * np.exp(-x ** 2 - y ** 2) 18 19 20 # n条等值线 21 n = 8 22 # 得到等值线 23 C = plt.contour(X, Y, f(X, Y), n, colors='black') 24 # 在等值线区域填充颜色 25 plt.contourf(X, Y, -f(X, Y), n, cmap=plt.cm.hot) 26 # 为等值线注明含义 27 plt.clabel(C, inline=1, fontsize=10) 28 plt.colorbar() 29 plt.show()
效果图：
