相関係数のヒートマップ表示
import pandas as pd import matplotlib.pyplot as plt import seaborn as sns plt.figure(figsize=(10,8)) sns.heatmap(df.corr(), vmin=-1.0, vmax=1.0, annot=True, cmap='coolwarm', linewidths=0.1)
* sns.heatmap()
→seabornライブラリのheatmap関数を呼出し
* vmin, vmaxオプション
→最小値と最大値の指定
* annot=Trueオプション
→色付きのセルの中に相関係数の数値を表示
* cmap, linewidthsオプション
→見た目を調整するためのそれぞれ塗り色のパターンとセル間の線の太さを指定
* 塗り色のパターンは color example code: colormaps_reference.py - Matplotlib 2.0.2 documentationから選択可能