Me gustaría trazar una matriz de 2 d desde numpy como una matriz coloreada en Matplotlib. Tengo la siguiente matriz 9-por-9:mostrando una matriz coloreada de 2d en matplotlib en Python
my_array = diag(ones(9))
# plot the array
pcolor(my_array)
me gustaría establecer los tres primeros elementos de la diagonal de ser un color determinado, el siguiente de tres a ser de un color diferente, y los tres últimos un color diferente Me gustaría especificar el color con una cadena de código hexadecimal, como "# FF8C00". ¿Cómo puedo hacer esto?
Además, ¿cómo puedo establecer el color de los elementos de 0 valores para pcolor?