Uzyskaj domyślny cykl kolorów linii

Zauważyłem, kiedy wykreślasz, że pierwsza linia jest niebieska, potem zielona, ​​potem czerwona i tak dalej.

Czy jest jakiś sposób, aby uzyskać dostęp do tej listy kolorów? Widziałem miliony postów o tym, jak zmienić cykl kolorów lub uzyskać dostęp do iteratora, ale nie o tym, jak uzyskać listę kolorów, które domyślnie przechodzą matplotlib.

W wersjach matplotlib> = 1.5 można wydrukować rcParamnazwany axes.prop_cycle:

print plt.rcParams['axes.prop_cycle'].by_key()['color']

# [u'#1f77b4', u'#ff7f0e', u'#2ca02c', u'#d62728', u'#9467bd', u'#8c564b', u'#e377c2', u'#7f7f7f', u'#bcbd22', u'#17becf']

Lub równoważnie w python3:

print(plt.rcParams['axes.prop_cycle'].by_key()['color'])

W wersjach <1.5 nazywało się to color_cycle:

print plt.rcParams['axes.color_cycle']

# [u'b', u'g', u'r', u'c', u'm', u'y', u'k']

Zwróć uwagę, że domyślny cykl kolorów zmienił się w wersji 2.0.0 http://matplotlib.org/users/dflt_style_changes.html#colors-in-default-property-cycle

Często nie ma potrzeby pobierania domyślnego cyklu kolorów z dowolnego miejsca, ponieważ jest to domyślny, więc wystarczy go użyć.

import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(111)

t = np.arange(5)

for i in range(4):
    line, = ax.plot(t,i*(t+1), linestyle = '-')
    ax.plot(t,i*(t+1)+.3,color = line.get_color(), linestyle = ':')

plt.show()

Jeśli chcesz użyć domyślnego cyklu kolorów do czegoś innego, istnieje oczywiście kilka opcji.

Mapa kolorów „tab10”

Najpierw należy wspomnieć, że "tab10"mapa kolorów zawiera kolory z domyślnego cyklu kolorów, można ją pobrać za pośrednictwem cmap = plt.get_cmap("tab10").

Odpowiednik powyższego byłby zatem

import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(111)

t = np.arange(5)
cmap = plt.get_cmap("tab10")
for i in range(4):
    ax.plot(t,i*(t+1),   color=cmap(i), linestyle = '-')
    ax.plot(t,i*(t+1)+.3,color=cmap(i), linestyle = ':')

plt.show()

Kolory z cyklu kolorów

Można również skorzystać z cyklicznych zmian kolorów bezpośrednio cycle = plt.rcParams['axes.prop_cycle'].by_key()['color']. Daje to listę z kolorami z cyklu, których możesz użyć do iteracji.

import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(111)

t = np.arange(5)
cycle = plt.rcParams['axes.prop_cycle'].by_key()['color']

for i in range(4):
    ax.plot(t,i*(t+1),   color=cycle[i], linestyle = '-')
    ax.plot(t,i*(t+1)+.3,color=cycle[i], linestyle = ':')

plt.show()

CNnotacja

Wreszcie, CNnotacja pozwala uzyskać Nth kolor cyklu kolorów color="C{}".format(i). Działa to jednak tylko dla pierwszych 10 kolorów ( N in [0,1,...9])

import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(111)

t = np.arange(5)

for i in range(4):
    ax.plot(t,i*(t+1),   color="C{}".format(i), linestyle = '-')
    ax.plot(t,i*(t+1)+.3,color="C{}".format(i), linestyle = ':')

plt.show()

Wszystkie przedstawione tutaj kody dają ten sam wykres.

Zrewidowano notację CN

Chciałbym zająć się nowym rozwojem Matplotlib. W poprzedniej odpowiedzi czytaliśmy

Wreszcie, CNnotacja pozwala uzyskać Nth kolor cyklu kolorów color="C{}".format(i). Działa to jednak tylko dla pierwszych 10 kolorów ( N in [0,1,...9])

ale

import numpy as np 
import matplotlib.pyplot as plt 

t = np.linspace(0,6.28, 629)                                                      
for N in (1, 2): 
    C0N, C1N = 'C%d'%(N), 'C%d'%(N+10) 
    plt.plot(t, N*np.sin(t), c=C0N, ls='-',  label='c='+C0N) 
    plt.plot(t, N*np.cos(t), c=C1N, ls='--', label='c='+C1N) 
plt.legend() ; plt.grid() ; plt.show()                                           

daje

jeśli szukasz szybkiego, jednowierszowego, aby uzyskać kolory RGB, których matplotlib używa dla swoich linii, oto jest:

>>> import matplotlib; print('\n'.join([str(matplotlib.colors.to_rgb(c)) for c in matplotlib.pyplot.rcParams['axes.prop_cycle'].by_key()['color']]))
(0.12156862745098039, 0.4666666666666667, 0.7058823529411765)
(1.0, 0.4980392156862745, 0.054901960784313725)
(0.17254901960784313, 0.6274509803921569, 0.17254901960784313)
(0.8392156862745098, 0.15294117647058825, 0.1568627450980392)
(0.5803921568627451, 0.403921568627451, 0.7411764705882353)
(0.5490196078431373, 0.33725490196078434, 0.29411764705882354)
(0.8901960784313725, 0.4666666666666667, 0.7607843137254902)
(0.4980392156862745, 0.4980392156862745, 0.4980392156862745)
(0.7372549019607844, 0.7411764705882353, 0.13333333333333333)
(0.09019607843137255, 0.7450980392156863, 0.8117647058823529)

Lub dla uint8:

import matplotlib; print('\n'.join([str(tuple(int(round(v*255)) for v in matplotlib.colors.to_rgb(c))) for c in matplotlib.pyplot.rcParams['axes.prop_cycle'].by_key()['color']]))
(31, 119, 180)
(255, 127, 14)
(44, 160, 44)
(214, 39, 40)
(148, 103, 189)
(140, 86, 75)
(227, 119, 194)
(127, 127, 127)
(188, 189, 34)
(23, 190, 207)