Verwenden Sie Titanic Victim Data von kaggle, um eine Liste der Korrelationskoeffizienten mithilfe der Heatmap anzuzeigen. Beachten Sie, dass Sie nur anhand eines numerischen Werts prüfen können
corr.py
# lib install
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
%matplotlib inline
train = pd.read_csv('./train.csv')
train.head()
corr.py
plt.figure(figsize=(8, 6)) #heatmap size
sns.heatmap(train.corr(), annot=True, cmap='plasma', linewidths=.5) # annot:Gibt an, ob die Wertlinienbreiten angezeigt werden sollen:Schnittlinie
In diesem Beispiel ist es wichtig zu überleben. Wenn Sie sich also die Spalte Überleben ansehen, finden Sie die korrelierten Variablen.
Referenz: Seaborn Heatmap
Recommended Posts