セクション 5. 章 7
single
ヒートマップ
メニューを表示するにはスワイプしてください
定義
ヒートマップは、各値の大きさを色で表現することで、二次元データを可視化する手法。
この例では、変数間のペアワイズ相関をヒートマップで可視化。
シンプルなヒートマップの作成
seaborn.heatmap() は2次元データセットを受け取る関数。一般的な用途としては、相関行列のプロットが挙げられる。DataFrame に対して .corr() を呼び出して相関を計算し、その結果の行列を heatmap() に渡す。
1234567891011import seaborn as sns import matplotlib.pyplot as plt import pandas as pd url = 'https://content-media-cdn.codefinity.com/courses/47339f29-4722-4e72-a0d4-6112c70ff738/countries_data.csv' countries_df = pd.read_csv(url, index_col=0) correlation_matrix = countries_df.corr(numeric_only=True) sns.heatmap(correlation_matrix) plt.show()
相関行列は数値列のみから作成される(numeric_only=True)。
注釈とカラーマップ
annot=True を設定すると、各セル内に相関値が表示されます。また、cmap を使ってカラーマップを選択できます。
ノート
ヒートマップの色は、cmap パラメータを設定することで変更できます(詳細は 「カラーパレットの選択」 記事を参照してください)。
1234567891011import seaborn as sns import matplotlib.pyplot as plt import pandas as pd url = 'https://content-media-cdn.codefinity.com/courses/47339f29-4722-4e72-a0d4-6112c70ff738/countries_data.csv' countries_df = pd.read_csv(url, index_col=0) correlation_matrix = countries_df.corr(numeric_only=True) sns.heatmap(correlation_matrix, annot=True, cmap='viridis') plt.show()
右側のカラーバーは、cbar=Falseを設定することで非表示にできます。
さらに学ぶ
ほとんどの場合、ヒートマップのカスタマイズで必要なのはこれだけですが、さらに詳しく知りたい場合は、 heatmap() documentationを参照してください。
可読性の向上
ヒートマップの可読性を高める最後のポイントは、すでにおなじみのxticks()やyticks()関数を使って目盛りを回転させることです:
1234567891011121314import seaborn as sns import matplotlib.pyplot as plt import pandas as pd url = 'https://content-media-cdn.codefinity.com/courses/47339f29-4722-4e72-a0d4-6112c70ff738/countries_data.csv' countries_df = pd.read_csv(url, index_col=0) correlation_matrix = countries_df.corr(numeric_only=True) sns.heatmap(correlation_matrix, annot=True, cmap='viridis') plt.xticks(rotation=20) plt.yticks(rotation=20) plt.show()
タスク
スワイプしてコーディングを開始
- 相関行列を作成するために正しいメソッドを使用。
- メソッドの引数を設定し、数値変数のみを含める。
- 正しい関数を使用してヒートマップを作成。
- 第一引数で
correlation_matrixをヒートマップのデータとして指定。 - 第二引数で行列の各セルに値を表示。
- 第三(最も右側の)引数でヒートマップのパレット(カラーマップ)を
'crest'に設定。 xticks()およびyticks()のキーワード引数で x 軸・y 軸の目盛りを反時計回りに15度回転。
解答
すべて明確でしたか?
フィードバックありがとうございます!
セクション 5. 章 7
single
AIに質問する
AIに質問する
何でも質問するか、提案された質問の1つを試してチャットを始めてください