【1行で】Pandasだけで相関行列をヒートマップっぽく可視化する

Pandasだけで相関行列をヒートマップっぽく可視化します。
MatplotlibもSeabornも使いません。

やり方

df.corr().style.background_gradient(axis=None)

具体例

scikit-learnに付属するBoston house-pricesのデータセットを使います。

import pandas as pd
from sklearn import datasets
boston = datasets.load_boston()
df = pd.DataFrame(boston['data'], columns=boston['feature_names'])
df.corr().style.background_gradient(axis=None)

カラーマップも使えます。

df.corr().style.background_gradient(axis=None, cmap='viridis')

ただし、カラーバーの範囲が[min, 1]ですので、[-1, 1]にするにはlow=(1 + df.corr().values.min()) / (1 - df.corr().values.min())を指定します。

df.corr().style.background_gradient(axis=None, cmap='viridis', low=(1 + df.corr().values.min()) / (1 - df.corr().values.min()))