Valeurs de corrélation dans pairplot()

Question

Existe-t-il un moyen d'afficher les valeurs de corrélation par paire avec seaborn.pairplot() comme dans l'exemple ci-dessous (créé avec ggpairs() en R ) ? Je peux faire les tracés en utilisant le code ci-joint, mais je ne peux pas ajouter les corrélations. Merci de votre compréhension.

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

iris = sns.load_dataset('iris')
g = sns.pairplot(iris, kind='scatter', diag_kind='kde')
# remove upper triangle plots
for i, j in zip(*np.triu_indices_from(g.axes, 1)):
    g.axes[i, j].set_visible(False)
plt.show()

Answer 1

Si vous utilisez PairGrid au lieu de pairplot vous pouvez passer une fonction personnalisée qui calculera le coefficient de corrélation et l'affichera sur le graphique :

from scipy.stats import pearsonr
def reg_coef(x,y,label=None,color=None,**kwargs):
    ax = plt.gca()
    r,p = pearsonr(x,y)
    ax.annotate('r = {:.2f}'.format(r), xy=(0.5,0.5), xycoords='axes fraction', ha='center')
    ax.set_axis_off()

iris = sns.load_dataset("iris")
g = sns.PairGrid(iris)
g.map_diag(sns.distplot)
g.map_lower(sns.regplot)
g.map_upper(reg_coef)