Python绘制相关系数热力图

Python 绘制相关系数热力图

这里用到的库有seaborn，pandas和matplotlib，使用jupyter notebook进行编译

import seaborn as sns
import pandas as pd
import matplotlib.pyplot as plt

df = pd.read_csv('data.csv')#这里的data.csv为任意csv文件
df_corr = df.corr()#计算相关性系数
plt.figure(figsize=(20,15))#figsize可以规定热力图大小
fig=sns.heatmap(df_corr,annot=True,fmt='.2g')#annot为热力图上显示数据；fmt='.2g'为数据保留两位有效数字
fig

fig.get_figure().savefig('df_corr.png')#保留图片

对于sns.heatmap()中的参数可以参考官网解释：
https://seaborn.pydata.org/generated/seaborn.heatmap.html?highlight=heatmap#seaborn.heatmap