python--seaborn热力图

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热力图