Tableau particulier...

Bonjour à toutes et à tous !

Voilà, je faisais un programme et Python (qui marche plutôt bien) qui me renvoie un tableau (une matrice quoi) et une liste associée à ce tableau.
En fait ce tableau ne contient que les valeurs 0; 1; 2 ou 3
Afin de visualiser le résultat, je préférerais que le programme me renvoie une tableau, dont chaque case serait d'une couleur que j'ai défini (je voulais associer le bleu à 0, le jaune à 1, le vert à 2, et le rouge à 3)

J'ai déjà cherché des tutos sur Internet, j'ai utilisé matplotlib, mais pour créer moi-même ma liste de couleurs, j'ai pas trouvé.
Et d'ailleurs, en petit bonus, je voudrais avoir une légende qui décrive à quoi correspondent les couleurs (mais avec du texte j'ai l'impression que ça marche pas bien.....)

Bref, merci à tous d'avance de bien vouloir me répondre ! =D