基本形

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の配列)