Wie man Heatmap mit Text in Matplotlib annotieren?

Question

ich mit einer Heatmap in matplotlib am Plotten:

plt.pcolor(rand(5,5))

wie kann ich die Heatmap mit den tatsächlichen Zahlen aufgetragen annotieren? Bezeichnen Sie in jeder Zelle der gezeichneten Heatmap den Wert, der dieser Zelle in der 5x5-Matrix entspricht, die an pcolor übergeben wird. Vielen Dank.

Answer 1

Es gibt keine automatische Funktion, so etwas zu tun, aber man konnte jeden Punkt Schleife durch und Text in der entsprechenden Stelle setzen:

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

Die Seaborn Heatmap erledigt den Job automatisch, indem er annot=True einstellt.

Siehe this für ein Beispiel.