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的相關系數矩陣

這樣就可以很清楚的看到特征之間的相關關系

如何快速繪制相關系數矩陣