Создать генератор цветов из заданной карты цветов в matplotlib - PullRequest
Новая
       10

Создать генератор цветов из заданной карты цветов в matplotlib

15 голосов
/ 10 июня 2010

У меня есть серия линий, каждая из которых должна быть нанесена отдельным цветом. Каждая строка на самом деле состоит из нескольких наборов данных (положительные, отрицательные области и т. Д.), Поэтому я хотел бы иметь возможность создавать генератор, который будет передавать один цвет за раз по спектру, например карту gist_rainbow показано здесь .

Я нашел следующие работы, но они кажутся очень сложными и, что более важно, трудными для запоминания, 

from pylab import *

NUM_COLORS = 22

mp = cm.datad['gist_rainbow']
get_color = matplotlib.colors.LinearSegmentedColormap.from_list(mp, colors=['r', 'b'], N=NUM_COLORS)
...
# Then in a for loop
    this_color = get_color(float(i)/NUM_COLORS)

Более того, он не охватывает диапазон цветов на карте gist_rainbow, мне нужно переопределить карту.

Может быть, генератор - не лучший способ сделать это, если так, каков принятый способ?

1 Ответ

25 голосов
/ 11 июня 2010

Для индексирования цветов из определенной цветовой карты вы можете использовать: 

import pylab
NUM_COLORS = 22

cm = pylab.get_cmap('gist_rainbow')
for i in range(NUM_COLORS):
    color = cm(1.*i/NUM_COLORS)  # color will now be an RGBA tuple

# or if you really want a generator:
cgen = (cm(1.*i/NUM_COLORS) for i in range(NUM_COLORS))
...