Pythonを使って、相関行列のヒートマップとPvaluesを入れたいです。

Question

実現したいこと

Pythonで、写真のようにheatmapに相関係数と相関係数のPvaluesを入れたいです。
なお、Pvaluesは、※を入れたいです。

https://gyazo.com/0a8952e18083b52c94374cb6bb158350

前提

ここに質問の内容を詳しく書いてください。
Pythonで、写真のようにheatmapに相関係数と相関係数のPvaluesを入れたいです。
なお、Pvaluesは、※を入れたいです。
一方、現在Pythonで相関係数を入れることはできてますが、Pvaluesの米印を
どうやって入れたらいいか、わからない状態です。
どなたか、ご教授いただけませんでしょうか？

該当のソースコード

# 必要なライブラリのインポート
import numpy as np
import pandas as pd
import seaborn as sns

np.random.seed(0)
# 10行5列の配列を生成
arr = np.random.randint(0,101,50).reshape(10, 5)
# データフレームの作成
col = ['国語', '数学', '社会', '理科', '英語']
df = pd.DataFrame(arr, columns=col)
df

# 相関行列の算出
corr = df.corr()
# ヒートマップの作成
sns.heatmap(corr)

sns.heatmap(corr, annot=True)

https://gyazo.com/0a8952e18083b52c94374cb6bb158350

Comments

[meg_]

リンクではなく画像として貼り付けしていただけませんか？

Answer 1

※ np.random.randint(0,101,50) の場合、P-value のグレードが全て NS になってしまうので、np.random.randint(0,3,50) に変更しています。

# 必要なライブラリのインポート
import numpy as np
import pandas as pd
import seaborn as sns
from scipy.stats import pearsonr
from matplotlib import pyplot as plt

plt.rcParams['font.family'] = 'MS Gothic'

np.random.seed(0)
# 10行5列の配列を生成
arr = np.random.randint(0, 3, 50).reshape(10, 5)

# データフレームの作成
col = ['国語', '数学', '社会', '理科', '英語']
df = pd.DataFrame(arr, columns=col)

# 相関行列の算出
corr = df.corr().round(3)
pval = df.corr(lambda x, y: pearsonr(x, y)[1])\
         .applymap(lambda p: '' if p > 0.05 else \
                             ''.join(['※' for v in [.05, .01, .001] if p <= v]))

# ヒートマップの作成
sns.heatmap(corr, annot=corr.astype(str) + '\n' + pval, fmt='s')
plt.show()

Comments

[Ryosuke0524]

ありがとうございます！非常に綺麗なグラフを書くことができました。ありがとうございました！！