Отримати кольоровий цикл ліній за замовчуванням


95

Я помітив, коли ви складаєте графік, що перший рядок синій, потім зелений, потім червоний тощо.

Чи є спосіб отримати доступ до цього списку кольорів? Я бачив мільйон публікацій про те, як змінити колірний цикл або отримати доступ до ітератора, але не про те, як просто отримати список кольорів, які matplotlib циклічно перетворює за замовчуванням.

Відповіді:


118

У версіях matplotlib> = 1.5 ви можете надрукувати rcParamвикликаний 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']

Або еквівалентно python3:

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

У версіях <1.5 це називалося color_cycle:

print plt.rcParams['axes.color_cycle']

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

Зверніть увагу, що кольоровий цикл за замовчуванням змінено у версії 2.0.0 http://matplotlib.org/users/dflt_style_changes.html#colors-in-default-property-cycle


3
Дякую! Незначна корекція: Першою має бути:lines_colour_cycle = [p['color'] for p in plt.rcParams['axes.prop_cycle']]
Пітер

4
@Peter, так, абоplt.rcParams['axes.prop_cycle'].by_key()['color']
tmdavison

91

Часто немає необхідності отримувати кольоровий цикл за замовчуванням з будь-якого місця, оскільки він є типовим, тому достатньо лише його використання.

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

введіть тут опис зображення

Якщо ви хочете використовувати кольоровий цикл за замовчуванням для чогось іншого, звичайно, є кілька варіантів.

карта кольорів "tab10"

Спочатку слід зазначити, що "tab10"кольорова карта включає кольори із кольорового циклу за замовчуванням, ви можете отримати її через cmap = plt.get_cmap("tab10").

Звідси еквівалент вищезазначеному

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

Кольори з кольорового циклу

Ви можете також використовувати колір зарядннік безпосередньо cycle = plt.rcParams['axes.prop_cycle'].by_key()['color']. Це дає список кольорів із циклу, які ви можете використовувати для перегляду.

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

CNнотація

Нарешті, CNпозначення дозволяє отримати Nй колір циклу кольору, color="C{}".format(i). Однак це працює лише для перших 10 кольорів ( 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()

Всі представлені тут коди дають однаковий сюжет.


чи можна сказати matplotlib використовувати якийсь цикл? Потрібність перебору масиву кольорів означає, що вам потрібно додати логіку повернення до індексу 0 після використання одного повного циклу.
Мехді

@Mehdi Звичайно, matplotlib використовує кольоровий цикл. Це питання вимагає отримання кольорів цього циклу.
ImportanceOfBeingErnest

5
Я думаю, що CNу вашій відповіді позначення повинні бути набагато помітнішими, я їх майже пропустив. Я підозрюю, що переважна більшість випадків використання задоволена можливістю отримати доступ лише до перших 10 кольорів, а передача 'C1'друзів - це набагато менший шаблон, ніж явне захоплення циклу реквізиту.
Андраш

3

Позначення CN переглянуто

Я хотів би звернутися до нової розробки Matplotlib. У попередній відповіді ми читали

Нарешті, CNпозначення дозволяє отримати Nй колір циклу кольору, color="C{}".format(i). Однак це працює лише для перших 10 кольорів ( N in [0,1,...9])

але

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

дає

введіть тут опис зображення


2

якщо ви шукаєте швидкий однокласник, щоб отримати кольори RGB, які matplotlib використовує для своїх ліній, ось він:

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

Або для 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)
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.