【Python, Heat map】相関係数を可視化する

##はじめに
変数が多くなると，相関係数表は見づらくなりますね．

そこで今回はSeabornのHeatmapを用いて可視化していきます．
タイタニックのデータを使っていて、一部加工してあります。

Heat map

実装

import seaborn as sns sns.heatmap(可視化したいデータ, 引数)
大雑把にはこんな感じです。

import seaborn as sns
cor = train_df[[ 'Age','Sex', 'Family','Pclass','Embarked','Survived', 'Fare']].corr()
sns.heatmap(cor, cmap= sns.color_palette('coolwarm', 10), annot=True,fmt='.2f', vmin = -1, vmax = 1)

だいぶ見やすくなりました！今回は引数にcmapとannot、fmtをとって、小数第2位までの相関係数を表示させました。
文字が切れているので、適宜サイズを調整してください。

良く使いそうな引数の紹介

• data: Pandasデータフレームの場合は行と列の名前が使われます．
• vmin, vmax: データの最大値と最小値を指定してくれます．
• cmap: 色の指定になります．10という数字は、10段階で色に段階をつけることを示しいています．今回は、相関係数を示したいので、Diverging Colormapから色を選びました．coolwarm以外には、bwr, seismicがあります．（ご指摘ありがとうございます！）matplotlibと色のコードは一緒です．以下のサイトに色とそれを示すコードが載ってます．cf. https://matplotlib.org/examples/color/colormaps_reference.html
• annot: Trueで数値を表示させます．指定しないとFalseとなり、値は表示されません．

その他の引数は参考サイトを参照してください．

参考サイト

・Seaborn でヒートマップを作成する　　
https://pythondatascience.plavox.info/seaborn/heatmap

間違ってる箇所等ありましたら、細かいことでも教えて頂けるとありがたいです！