使用matplotlib繪制熱圖(heatmap)全過程

更新時間：2022年12月17日 12:18:52 作者：coder_Gray

這篇文章主要介紹了使用matplotlib繪制熱圖(heatmap)全過程，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

matplotlib繪制熱圖(heatmap)

python常用的繪圖庫就是matplotlib，今天在給公司繪圖時，偶然間發(fā)現(xiàn)matplotlib可以繪制熱圖，并且十分簡潔。

拿出來跟大家分享一下

（由于涉及到公司數(shù)據(jù)問題，這里采用隨機數(shù)生成數(shù)據(jù)進行實驗）

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']
 
    #準備數(shù)據(jù)階段，利用random生成二維數(shù)據(jù)（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()