seaborn heatmap
概要
- seabornでヒートマップを描画する
- 相関行列を可視化するのに便利
サンプルコード(株価の相関行列を可視化)
import seaborn as sns
import matplotlib.pyplot as plt
pivot = df.pivot(index=["date"], columns=["ticker"], values=["Close"]).reset_index(drop=True)
pivot.columns = pivot.columns.droplevel()
sns.heatmap(pivot.corr(), vmin=0.8, vmax=1, cmap="Greens")
pivotは日付をindex、銘柄をcolumnsに持つDataFramepivot.corr()で相関行列を計算sns.heatmapでヒートマップを描画vminとvmaxでカラーバーの最小値と最大値を指定cmapでカラーマップを指定
