jak opisać termogram z tekstem w matplotlib?

Question

ja wykreślenie mapę cieplną w matplotlib używając:

plt.pcolor(rand(5,5))

jak mogę adnotację na mapę cieplną z rzeczywistych liczb kreślone? w każdej komórce wykreślonej mapy cieplnej należy umieścić wartość odpowiadającą tej komórce w macierzy 5x5 przekazaną do pcolor. dzięki.

Answer 1

Nie ma funkcja automatycznego zrobić coś takiego, ale można pętli każdego punktu i umieścić tekst w odpowiednim miejscu:

import matplotlib.pyplot as plt 
import numpy as np 

data = np.random.rand(5, 4) 
heatmap = plt.pcolor(data) 

for y in range(data.shape[0]): 
    for x in range(data.shape[1]): 
     plt.text(x + 0.5, y + 0.5, '%.4f' % data[y, x], 
       horizontalalignment='center', 
       verticalalignment='center', 
       ) 

plt.colorbar(heatmap) 

plt.show() 

HTH

Answer 2

Seaborn HeatMap wykonuje zadanie automatycznie, ustawiając annot=True.

Przykładem jest this.