【Python 热力图绘制】傻瓜式教学:最常用的几个API

效果展示

本例是绘制如下效果的热力图的代码示例:

在这里插入图片描述

使用

热力图的绘制依赖于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对比下就知道了)

猜你喜欢

转载自blog.csdn.net/weixin_39274659/article/details/107235625
今日推荐