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

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

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

matplotlib繪制熱圖(heatmap)

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

拿出來跟大家分享一下

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

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():
    #定義熱圖的橫縱坐標(biāo)
    xLabel = ['A','B','C','D','E']
    yLabel = ['1','2','3','4','5']
 
    #準(zhǔn)備數(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個(gè)劃分，并在第1個(gè)位置上進(jìn)行作圖
    ax = fig.add_subplot(111)
    #定義橫縱坐標(biāo)的刻度
    ax.set_yticks(range(len(yLabel)))
    ax.set_yticklabels(yLabel, fontproperties=font)
    ax.set_xticks(range(len(xLabel)))
    ax.set_xticklabels(xLabel)
    #作圖并選擇熱圖的顏色填充風(fēng)格，這里選擇hot
    im = ax.imshow(data, cmap=plt.cm.hot_r)
    #增加右側(cè)的顏色刻度條
    plt.colorbar(im)
    #增加標(biāo)題
    plt.title("This is a title", fontproperties=font)
    #show
    plt.show()
 
d = draw()