本例是绘制如下效果的热力图的代码示例:
热力图的绘制依赖于seaborn库, 可以通过 pip install seaborn
下载。
假设已经拥有了一个2维矩阵G,矩阵的每个元素对应于热力图的每个格子, 那么就可以用以前代码:
import seaborn as sns
import matplotlib.pyplot as plt
fig, ax = plt.subplots(1,1) #必须这一句,不然无法show
sns.heatmap(G, camp = 'Blues', annot = True)
plt.show()
短短三行, 就能画出上面展示的热力图了。
其实就第二句需要解释一下:
sns.heatmap(G, camp = 'Blues', annot = True)