Utilisez Kaggle's Titanic Victim Data pour afficher une liste de coefficients de corrélation à l'aide d'une carte thermique. Notez que vous ne pouvez vérifier que par valeur numérique
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:Afficher ou non les largeurs de ligne de valeur:Ligne de coupe
Dans cet exemple, il est important que vous ayez survécu, afin que vous puissiez trouver les variables corrélées en regardant la colonne Survived.
Référence: Seaborn Heatmap
Recommended Posts