Настройка цветов в matplotlib - тепловая карта
Как я могу указать цвета в карте тепла. В этом примере данные являются одним из 4 значений.{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()
Как я могу определить эти цвета таким образом, чтобы представлять цвета = {0: «зеленый», 1: «красный», 2: «черный», 3: «желтый»}