Python绘制相关性热力图

[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

保存图片：

fig.get_figure().savefig('df_corr.png',bbox_inches='tight',transparent=True)#保存图片
#bbox_inches让图片显示完整，transparent=True让图片背景透明