seaborn是专门用于统计数据可视化的包，可媲美R语言中的ggplot2包。本文介绍用seaborn绘制热力图。
环境
- python3.9
- win10 64bit
- seaborn==0.11.1
- matplotlib==3.3.4
- pandas==1.2.1
数据
热力图的一个常见应用场景是绘制相关系数热力图，数据准备一个相关系数矩阵。
import seaborn as sns
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
plt.rcParams['figure.dpi'] = 80 # 图形分辨率
pd.options.display.notebook_repr_html=False # 表格显示
# 相关系数矩阵
np.random.seed(1) # 随机种子
mat=pd.DataFrame(np.random.rand(3,6),columns=list('abcdef')).corr()
mat
a b c d e f
a 1.000000 0.297407 -0.610065 -0.908297 -0.997893 -0.991160
b 0.297407 1.000000 -0.937936 0.129262 -0.234835 -0.421449
c -0.610065 -0.937936 1.000000 0.222660 0.557370 0.709796
d -0.908297 0.129262 0.222660 1.000000 0.933525 0.844767
e -0.997893 -0.234835 0.557370 0.933525 1.000000 0.980463
f -0.991160 -0.421449 0.709796 0.844767 0.980463 1.000000
绘制
调用heatmap方法绘制热力图。
# 热力图
sns.heatmap(mat)
plt.show()
heatmap_6_0.png
调整调色板
设置vmin和vmax参数可以调整调色板的下限值和上限值。
# 设置调色板上下限值
sns.heatmap(mat,vmin=0,vmax=1)
plt.show()
heatmap_8_0.png
设置cmap参数，可以修改调色板样式。
# 修改调色板
sns.heatmap(mat,cmap='YlGnBu')
plt.show()
heatmap_10_0.png
隐藏图例
设置参数cbar=False可以隐藏图例。
# 隐藏图例
sns.heatmap(mat,cbar=False)
plt.show()
heatmap_12_0.png
显示数值
设置参数annot=True可以显示热力图上的具体数值，设置fmt参数，可以修改数值显示的样式。
# 显示数值热力图
sns.heatmap(mat,annot=True,fmt='.3f')
plt.show()
heatmap_14_0.png
更多参考seaborn热力图
