Matplotlib: пользовательская цветовая карта с тремя цветами

Question

Я хотел бы передать цветовую карту matplotlib.imshow построить массив NxN, где каждая ячейка может принимать только значения -1,0 и 1. Я хотел бы, чтобы моя карта цветов отображала ровно один цвет (в шестнадцатеричном коде) для каждого значения.

Я просто не знаю, как определить собственную цветовую карту, которая не является градацией между двумя цветами.

Answer 1

Решение очень похожего вопроса было предоставлено здесь. В итоге,colors.ListedColormap(list-of-colours), твой друг.

Answer 2

Нашел ответ на мой вопрос, как предположил Фернандо:

import matplotlib
colorsList = [(tuple rgb color 1),(tuple rgb color 2),(tuple rgb color 3)]
CustomCmap = matplotlib.colors.ListedColormap(colorsList)