seaborn heatmap

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に持つDataFrame
  • pivot.corr()で相関行列を計算
  • sns.heatmapでヒートマップを描画
    • vminとvmaxでカラーバーの最小値と最大値を指定
    • cmapでカラーマップを指定
heatmap