基本形
def draw_heatmap(data, min_value, max_value):
"""
:param data:二次元配列.入力するデータ
:param min_value:カラーバーの最小値
:param max_value:カラーバーの最大値
"""
# 描画する
fig, ax = plt.subplots()
heatmap = ax.pcolor(data, cmap=plt.cm.Blues, vmin=min_value, max_value=4)
ax.set_xticks(np.arange(data.shape[0])+0.5, minor=False)
ax.set_yticks(np.arange(data.shape[1])+0.5, minor=False)
ax.invert_yaxis()
ax.xaxis.tick_top()
plt.show()
カラーバーの範囲設定
-
vmin
とvmax
で設定 ### 色相の設定 -
cmap
のところ.以前のmatplotlibで色をグラデーション的に選択を参照
目盛りを真ん中に置く
-
ax.set_x(y)ticks
のところ - 普通のままだと端っこに目盛りがついちゃう
左上を原点にしたい
- ヒートマップを使うとき,左上から右下に流した方がいいときもある
-
ax.invert_yaxis()
にてy軸を逆転 -
ax.xaxis.tick_top
にて,x軸の値を上に持ってこれる
出来上がり
こんな感じ(4*4の配列)