开发笔记:如何快速绘制相关系数矩阵

import pandas as pd
train = pd.read_csv("dftest/train.csv")
import seaborn as sns
import matplotlib.pyplot as plt
def showcov(df):
dfData = df.corr()
plt.subplots(figsize=(9, 9)) # 设置画面大小
sns.heatmap(dfData, annot=True, vmax=1, square=True, cmap="Blues")
plt.savefig(‘./BluesStateRelation.png‘)
plt.show()
showcov(train)

df.corr()就是得到这个dataframe的相关系数矩阵

这样就可以很清楚的看到特征之间的相关关系