Python绘制相关性热力图

发布时间 2023-12-07 11:32:49作者: ministep88

[Python绘制相关性热力图 - 知乎](https://zhuanlan.zhihu.com/p/364624304)

随便一个数据就可

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
plt.rcParams['font.sans-serif'] = ['Arial Unicode MS']#让中文的地方显示出来

导入文件

df = pd.read_excel(r'C:\Users\XC\Desktop\coor.xlsx')
df.head()

看下结果:

求一下相关性

df_coor=df.corr()
df_coor.head()

看下结果:

设置画布,并画图:

plt.subplots(figsize=(9,9),dpi=1080,facecolor='w')# 设置画布大小,分辨率,和底色

fig=sns.heatmap(df_coor,annot=True, vmax=1, square=True, cmap="Blues", fmt='.2g')#annot为热力图上显示数据;fmt='.2g'为数据保留两位有效数字,square呈现正方形,vmax最大值为1
fig

保存图片: