• home
  • about
  • 全ての投稿
  • ソフトウェア・ハードウェアの設定のまとめ
  • 分析関連のまとめ
  • ヘルスケア関連のまとめ
  • 生涯学習関連のまとめ

seaborn heatmap

date: 2024-03-04 excerpt: seaborn heatmap

tag: seabornpythonmatplotlib可視化


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


seabornpythonmatplotlib可視化 Share Tweet