python Scatter plotの図の右上に相関係数を載せたい。

Question

実現したいこと

相関関係を示すために、pairplotでplot図を書きましたが、それだけだとわからないので、図の右上に相関係数を提示したいです。

発生している問題・エラーメッセージ

pairplotで一斉にグラフを輩出しましたが、これだけだとプロットだけで、
相関係数は出ませんでした。図の右上にそれぞれ相関係数が出れば嬉しいです。

該当のソースコード

import numpy as np
import pandas as pd
from sklearn.datasets import load_boston

data = load_boston()
df = pd.DataFrame(data=data.data, columns=data.feature_names)
df['TARGET'] = pd.DataFrame(data=data.target)
print(df.head())  # 初めの5レコードを表示

df2 = df.loc[:,['RM', 'RAD', 'LSTAT', 'TARGET']]
import matplotlib.pyplot as plt
import seaborn as sns

sns.pairplot(df2)
plt.show()

試したこと

pairplotで図を一斉に描きましたが、それぞれの図に相関係数を一気に書く方法があれば
教えていただけませんでしょうか？
また、p値も一緒に乗せられるとなお嬉しいです。

Answer 1

seaborn.FacetGrid.map — seaborn 0.12.2 documentation で各プロットに相関係数と p-value を描画(annotate)します。

import matplotlib.pyplot as plt
import seaborn as sns
from scipy.stats import pearsonr

def corrfunc(x, y, **kwargs):
    corr, pval = pearsonr(x, y)
    plt.gca().annotate(f'{corr:.3f}', xy=(.7, .9), xycoords=plt.gca().transAxes, color='red')
    plt.gca().annotate(f'{pval:.3e}', xy=(.7, .8), xycoords=plt.gca().transAxes, color='blue')

pair = sns.pairplot(df2)
pair.map(corrfunc)

plt.show()

Comments

[Ryosuke0524]

非常に綺麗な図を描くことができました。また相関係数やP値も載せることができました。
こちら、欠損値があるデータの場合は使えないのでしょうか？

[melian]

欠損値がある場合は、corr, pval = pearsonr(x, y) を以下の様に変更してみてください。(np.nan_to_num() で欠損値を 0 に置き換えます)

import numpy as np
corr, pval = pearsonr(np.nan_to_num(x), np.nan_to_num(y))