效果展示
本例是绘制如下效果的热力图的代码示例:
热力图的绘制依赖于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)
- G:用于绘图的2维矩阵数据,和热力图的每个格子的值一一对应
- camp:颜色风格, python自带了很多风格,我喜欢蓝色的’Blues’,可以百度搜索,选择自己心仪的。
- annot: 指定为True时,热力图的格子上会显示数字(不懂的把这个设为False对比下就知道了)