Uzyskaj stan cyklu kolorów matplotlib

Uwaga: w najnowszych wersjach matplotlib (>= 1.5) _get_lines zmieniło się. Teraz musisz użyć next(ax._get_lines.prop_cycler)['color'] w Pythonie 2 lub 3 (lub ax._get_lines.prop_cycler.next()['color'] W Python 2 ), aby uzyskać następny kolor z cyklu kolorów.

W miarę możliwości użyj bardziej bezpośredniego podejścia pokazanego w dolnej części odpowiedzi @ joe-Kinga. Ponieważ _get_lines nie jest związane z API, może się zmienić w przyszłości w sposób niezgodny wstecz.

Oto sposób, który działa w 1.5, który miejmy nadzieję będzie przyszłościowy, ponieważ nie opiera się na metodach poprzedzonych podkreślnikami:

colors = plt.rcParams["axes.prop_cycle"].by_key()["color"]

To da ci listę kolorów zdefiniowanych w kolejności dla obecnego stylu.

Jasne, to wystarczy.

#rainbow

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0,2*np.pi)
ax= plt.subplot(1,1,1)
ax.plot(np.sin(x))
ax.plot(np.cos(x))

rainbow = ax._get_lines.color_cycle
print rainbow
for i, color in enumerate(rainbow):
    if i<10:
        print color,

Daje:

<itertools.cycle object at 0x034CB288>
r c m y k b g r c m

Oto funkcja itertools, której matplotlib używa itertools.cykl

Edit: dzięki za komentarz, wygląda na to, że nie można skopiować iteratora. Pomysłem byłoby zrzucenie pełnego cyklu i śledzenie, jakiej wartości używasz, pozwól mi wrócić do tego.

Edit2: w porządku, to da ci następny kolor i stworzy nowy iterator, który zachowuje się tak, jakby next nie został wywołany. To nie zachowaj kolejność kolorowania, tylko następną wartość koloru, zostawiam to tobie.

Daje to następujące wyjście, zauważ, że stromość wykresu odpowiada indeksowi, np. pierwsze g jest najniższym wykresem i tak dalej.

#rainbow

import matplotlib.pyplot as plt
import numpy as np
import collections
import itertools

x = np.linspace(0,2*np.pi)
ax= plt.subplot(1,1,1)


def create_rainbow():
    rainbow = [ax._get_lines.color_cycle.next()]
    while True:
        nextval = ax._get_lines.color_cycle.next()
        if nextval not in rainbow:
            rainbow.append(nextval)
        else:
            return rainbow

def next_color(axis_handle=ax):
    rainbow = create_rainbow()
    double_rainbow = collections.deque(rainbow)
    nextval = ax._get_lines.color_cycle.next()
    double_rainbow.rotate(-1)
    return nextval, itertools.cycle(double_rainbow)


for i in range(1,10):
    nextval, ax._get_lines.color_cycle = next_color(ax)
    print "Next color is: ", nextval
    ax.plot(i*(x))


plt.savefig("SO_rotate_color.png")
plt.show()

Konsola

Next color is:  g
Next color is:  c
Next color is:  y
Next color is:  b
Next color is:  r
Next color is:  m
Next color is:  k
Next color is:  g
Next color is:  c

Chcę tylko dodać to, co @ Andi powiedział powyżej. Ponieważ color_cycle jest przestarzały w matplotlib 1.5, musisz użyć prop_cycler, jednak rozwiązanie Andi (ax._get_lines.prop_cycler.next()['color']) zwróciło mi ten błąd:

AttributeError: 'itertools.cykl 'obiekt nie posiada atrybutu 'next'

Kod, który działał dla mnie to: next(ax._get_lines.prop_cycler), co w rzeczywistości nie jest dalekie od oryginalnej odpowiedzi @ joe-kington.

Osobiście natknąłem się na ten problem podczas tworzenia osi twinx (), która resetuje kolor cycler. Potrzebowałem sposobu na poprawny cykl kolorów, ponieważ używałem style.use('ggplot'). Może być łatwiejszy / lepszy sposób, aby to zrobić, więc nie krępuj się mnie poprawić.

Ponieważ matplotlib używa itertools.cycle, możemy przejrzeć cały cykl kolorów, a następnie przywrócić iterator do poprzedniego stanu:

def list_from_cycle(cycle):
    first = next(cycle)
    result = [first]
    for current in cycle:
        if current == first:
            break
        result.append(current)

    # Reset iterator state:
    for current in cycle:
        if current == result[-1]:
            break
    return result

To powinno zwrócić listę bez zmiany stanu iteratora.

Użyj go z matplotlib >= 1.5:

>>> list_from_cycle(ax._get_lines.prop_cycler)
[{'color': 'r'}, {'color': 'g'}, {'color': 'b'}]

Lub z matplotlib

>>> list_from_cycle(ax._get_lines.color_cycle)
['r', 'g', 'b']