Anpassen von Farben in matplotlib - heatmap
Wie kann ich Farben in der Heatmap festlegen? In diesem Beispiel sind die Daten eindeutig einer von 4 Werten{0,1,2,3}
Index= ['aaa', 'bbb', 'ccc', 'ddd', 'eee']
Cols = ['A', 'B', 'C', 'D']
data= [[ 0, 3, 1, 1],[ 0, 1, 1, 1],[ 0, 1, 2, 1],[ 0, 2, 1, 2],[ 0, 1, 1, 1]]
print data
df = pd.DataFrame(data, index=Index, columns=Cols)
heatmap = plt.pcolor(np.array(data))
plt.colorbar(heatmap)
plt.show()
Wie kann ich diese Farben so angeben, dass sie Farben darstellen? = {0: 'grün', 1: 'rot', 2: 'schwarz', 3: 'gelb'}