Python Heatmap Grundstück colorbar

Question

ich auf einem Grundstück arbeite, die so weit wie folgt aussieht:

Und ich den folgenden Code verwende:

import matplotlib.pyplot as plt 
import numpy as np 
column_labels = ['A','B'] 
row_labels = ['A','B'] 
data = np.random.rand(2,2) 
print(type(data)) 
data = np.array([[1, 0.232], [0.119, 1]]) 
fig, ax = plt.subplots() 
heatmap = ax.pcolor(data, cmap=plt.cm.Blues) 

# put the major ticks at the middle of each cell 
ax.set_xticks(np.arange(data.shape[0])+0.5, minor=False) 
ax.set_yticks(np.arange(data.shape[1])+0.5, minor=False) 

# want a more natural, table-like display 
ax.invert_yaxis() 
ax.xaxis.tick_top() 

ax.set_xticklabels(row_labels, minor=False) 
ax.set_yticklabels(column_labels, minor=False) 
plt.show() 

Was ich zu bekommen versuche ist so etwas wie dieses:

Wo ich den Farbbalken sowie den Wert der Zellen auf ihnen sehen kann. Was müsste ich meinem Code hinzufügen, um diese zu erhalten?

Dank

Answer 1

Haben Sie in Seaborn untersucht?

import seaborn as sns 
data = np.random.rand(2,2) 
sns.heatmap(data, annot=True, linewidths=.5) 

Answer 2

Für die Farbleiste finden this example. Sie haben richtig geraten, der richtige Funktionsaufruf ist nur ax.colorbar().

In Bezug auf die Werte auf dem pcolor Grundstück, ich fürchte, Sie müssen sie "manuell" mit ax.text() hinzufügen. Berechnen und durchlaufen Sie die Zellenmittelwerte und rufen Sie die Funktion text() auf. Sie können die Schlüsselwortargumente horizontalalignment='center' und verticalalignment='center' verwenden, um den Textwert um x, y zentriert zu halten.