Utwórz generator kolorów z podanej mapy kolorów w matplotlib

Question

Mam serię linii, które należy narysować z osobnym kolorem. Każda linia składa się w rzeczywistości z kilku zestawów danych (dodatnich, negatywnych regionów itp.), Dlatego chciałbym móc stworzyć generator, który będzie zasilał jeden kolor na raz w całym spektrum, na przykład gist_rainbow mapa shown here.

Znalazłem następujące prace, ale wydaje się bardzo skomplikowane, a co ważniejsze, trudne do zapamiętania,

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) 

Ponadto nie obejmuje zakres kolorów w mapie gist_rainbow mam przedefiniować mapę .

Może generator nie jest najlepszym sposobem, aby to zrobić, jeśli tak, to w jaki sposób jest akceptowany?

Answer 1

Aby kolorach indeks z konkretnym colormap można użyć:

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))