seabornでヒートマップを作成する

ライブラリを読み込む

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns #データ可視化ライブラリ

二次元配列データを入力する

test=[[0, 1, 2], [3, 4, 5]] #二次元配列データ
test=pd.DataFrame(data=test)

ヒートマップを作成する

bbox_inches=”tight”でオブジェクトが描画されている場所に合わせて画像を出力し、pad_inches=0で画像の周囲の余白を削除します。

sns.heatmap(test, square=True) #ヒートマップを作成
plt.savefig("test.png", bbox_inches="tight", pad_inches=0) #ファイル名を指定