Цветная карта matplotlib - предопределяет сопоставление значений?

Question

У меня есть массив, который я просматриваю, используя imshow(). (imsave() действительно, но процесс должен быть идентичным).

Я знаю, что значения в массиве будут находиться в диапазоне от 0 до 9 и задаются вопросом, можно ли использовать cmap для установки каждого вывода на определенный «цвет». Возможно, сопоставив их с dict?

Answer 1

Просто используйте ListedColormap.

Как быстро (но уродливый), например:

import matplotlib.pyplot as plt 
from matplotlib.colors import ListedColormap 

cmap = ListedColormap(['red', 'green', 'blue', 'black'], 'indexed') 

fig, ax = plt.subplots() 
im = ax.imshow([range(4)], interpolation='none', cmap=cmap) 
fig.colorbar(im) 
plt.show()