如何快速绘制相关系数矩阵

如何快速绘制相关系数矩阵如何快速绘制相关系数矩阵

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的相关系数矩阵

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

猜你喜欢

转载自blog.51cto.com/yixianwei/2156109