Определение дискретной карты цветов для imshow в matplotlib

Question

У меня есть простое изображение, которое я показываю с помощью imshow в matplotlib.Я хотел бы применить пользовательскую цветовую карту, чтобы значения от 0 до 5 были белыми, от 5 до 10 - красными (очень простые цвета) и т. Д. Я пробовал следовать этому уроку:

http://assorted -experience.blogspot.com / 2007/07 / custom-colormaps.html со следующим кодом:

cdict = {
'red'  :  ((0., 0., 0.), (0.5, 0.25, 0.25), (1., 1., 1.)),
'green':  ((0., 1., 1.), (0.7, 0.0, 0.5), (1., 1., 1.)),
'blue' :  ((0., 1., 1.), (0.5, 0.0, 0.0), (1., 1., 1.))
}

my_cmap = mpl.colors.LinearSegmentedColormap('my_colormap', cdict, 3)

plt.imshow(num_stars, extent=(min(x), max(x), min(y), max(y)), cmap=my_cmap)
plt.show()

Но в итоге отображаются странные цвета, а мне нужно только 3-4цвета, которые я хочу определить.Как мне это сделать?

Answer 1

Вы можете использовать ListedColormap, чтобы указать белый и красный цвета как единственные цвета на цветовой карте, а границы определяют, где происходит переход от одного цвета к следующему:

import matplotlib.pyplot as plt
from matplotlib import colors
import numpy as np

np.random.seed(101)
zvals = np.random.rand(100, 100) * 10

# make a color map of fixed colors
cmap = colors.ListedColormap(['white', 'red'])
bounds=[0,5,10]
norm = colors.BoundaryNorm(bounds, cmap.N)

# tell imshow about color map so that only set colors are used
img = plt.imshow(zvals, interpolation='nearest', origin='lower',
                    cmap=cmap, norm=norm)

# make a color bar
plt.colorbar(img, cmap=cmap, norm=norm, boundaries=bounds, ticks=[0, 5, 10])

plt.savefig('redwhite.png')
plt.show()

Полученная фигура имеет только два цвета:

Я предложил по существу то же самое для несколько иного вопроса: Визуализация данных 2D-сетки в Python

Решение основано на примере matplotlib . В примере объясняется, что bounds должно быть на единицу больше, чем количество используемых цветов.

BoundaryNorm - это нормализация, которая отображает ряд значений в целые числа, которые затем используются для назначения соответствующих цветов. cmap.N, в приведенном выше примере, просто определяет количество цветов.

Answer 2

почему LinearSegmentedColormap показывает странный цвет, я думаю, эта ссылка будет полезна.

http://matplotlib.org/examples/pylab_examples/custom_cmap.html