import random
from matplotlib import pyplot as plt
from matplotlib import cm
from matplotlib import axes
from matplotlib.font_manager import FontProperties
font = FontProperties(fname='/Library/Fonts/Songti.ttc')
 
def draw():
    #定義熱圖的橫縱坐標
    xLabel = ['A','B','C','D','E']
    yLabel = ['1','2','3','4','5']
 
    #準備數據階段，利用random生成二維數據（5*5）
    data = []
    for i in range(5):
        temp = []
        for j in range(5):
            k = random.randint(0,100)
            temp.append(k)
        data.append(temp)
 
    #作圖階段
    fig = plt.figure()
    #定義畫布為1*1個劃分，并在第1個位置上進行作圖
    ax = fig.add_subplot(111)
    #定義橫縱坐標的刻度
    ax.set_yticks(range(len(yLabel)))
    ax.set_yticklabels(yLabel, fontproperties=font)
    ax.set_xticks(range(len(xLabel)))
    ax.set_xticklabels(xLabel)
    #作圖并選擇熱圖的顏色填充風格，這里選擇hot
    im = ax.imshow(data, cmap=plt.cm.hot_r)
    #增加右側的顏色刻度條
    plt.colorbar(im)
    #增加標題
    plt.title("This is a title", fontproperties=font)
    #show
    plt.show()
 
d = draw()

im = ax.imshow(data, cmap=plt.cm.hot_r)