Использование matplotlib для рисования цветовой шкалы с различимыми и непрерывными цветами

Question

Цветная полоса на рисунке ниже нарисована в matlab.Мне интересно, можно ли нарисовать подобную цветную полосу в python / matplotlib?Мне нужны несглаживающие изменения цвета в цветовой шкале, чтобы значения 0-5 были представлены в очень различимых и непрерывных цветахЕсли возможно, не могли бы вы дать мне пример?Спасибо ~

Answer 1

Использование ListedColormap и BoundaryNorm :

import numpy as np
import matplotlib as mpl
from matplotlib import pyplot
data = np.floor(np.random.random((10,10)) * 6)
cmap = mpl.colors.ListedColormap(['w', 'b', 'g', 'y', '#ff8c00', 'r'])
norm = mpl.colors.BoundaryNorm([0,1,2,3,4,5,6], cmap.N)
pyplot.imshow(data, cmap=cmap, norm=norm, interpolation='none')
pyplot.colorbar()
pyplot.show()

Некоторые примеры использования ListedColormap: цветные полосы , разноцветные линии .