绘制heatmap(热力图)

发布时间 2023-08-28 13:06:27作者: SL霸霸

什么是heatmap?

热力图在实际中常用于展示一组变量的相关系数矩阵,在展示列联表的数据分布上也有较大的用途,通过热力图我们可以非常直观地感受到数值大小的差异状况。

如何绘制heatmap图

1. 使用matplotlib库绘制

2. 使用seaborn库绘制

简单暴力,直接使用sns.heatmap(). 详细可查看[https://seaborn.pydata.org/generated/seaborn.heatmap.html#seaborn.heatmap]

# Annotated heatmaps 带注释的热力图
import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd 
import numpy as np 
sns.set_theme()

data = pd.read_csv('man.csv')

# 创建相关矩阵
corr_matrix = np.corrcoef(data, rowvar=False)

# Draw a heatmap with the numeric values in each cell
f, ax = plt.subplots(figsize=(9, 6))
sns.heatmap(corr_matrix, annot=True, fmt=".3f", linewidths=.5, ax=ax, center=1)
plt.show()

image

3.使用excel表绘制