Jak wybrać nowy kolor dla każdej kreślonej linii w figurze w matplotlib?

Chciałbym NIE określać koloru dla każdej kreślonej linii i aby każda linia miała inny kolor. Ale jeśli biegnę:

from matplotlib import pyplot as plt
for i in range(20):
    plt.plot([0, 1], [i, i])

plt.show()

wtedy otrzymuję ten wynik:

Jeśli spojrzysz na powyższy obrazek, zobaczysz, że matplotlib próbuje wybrać kolory dla każdej linii, które są różne, ale ostatecznie ponownie wykorzystuje kolory - dziesięć górnych linii używa tych samych kolorów, co dziesięć dolnych. Chcę po prostu powstrzymać go od powtarzania już używanych kolorów ORAZ / LUB podać listę kolorów do użycia.

matplotlib 1.5+

Możesz użyć axes.set_prop_cycle( przykład ).

matplotlib 1.0-1.4

Możesz użyć axes.set_color_cycle( przykład ).

matplotlib 0.x

Możesz użyć Axes.set_default_color_cycle.

Zwykle używam drugiego z nich:

from matplotlib.pyplot import cm
import numpy as np

#variable n below should be number of curves to plot

#version 1:

color=cm.rainbow(np.linspace(0,1,n))
for i,c in zip(range(n),color):
   plt.plot(x, y,c=c)

#or version 2:

color=iter(cm.rainbow(np.linspace(0,1,n)))
for i in range(n):
   c=next(color)
   plt.plot(x, y,c=c)

Przykład 2:

prop_cycle

color_cyclezostał wycofany w wersji 1.5 na korzyść tego uogólnienia: http://matplotlib.org/users/whats_new.html#added-axes-prop-cycle-key-to-rcparams

# cycler is a separate package extracted from matplotlib.
from cycler import cycler
import matplotlib.pyplot as plt

plt.rc('axes', prop_cycle=(cycler('color', ['r', 'g', 'b'])))
plt.plot([1, 2])
plt.plot([2, 3])
plt.plot([3, 4])
plt.plot([4, 5])
plt.plot([5, 6])
plt.show()

Pokazano również w (teraz źle nazwanym) przykładzie: http://matplotlib.org/1.5.1/examples/color/color_cycle_demo.html wspomnianym na: https://stackoverflow.com/a/4971431/895245

Przetestowano w matplotlib 1.5.1.

Nie wiem, czy możesz automatycznie zmienić kolor, ale możesz wykorzystać swoją pętlę, aby wygenerować różne kolory:

for i in range(20):
   ax1.plot(x, y, color = (0, i / 20.0, 0, 1)

W takim przypadku kolory będą się różnić od czarnego do 100% zielonego, ale możesz je dostroić, jeśli chcesz.

Zobacz dokumentację matplotlib plot () i poszukaj colorargumentu słowa kluczowego.

Jeśli chcesz podać listę kolorów, po prostu upewnij się, że masz wystarczająco dużą listę, a następnie użyj indeksu pętli, aby wybrać kolor

colors = ['r', 'b', ...., 'w']

for i in range(20):
   ax1.plot(x, y, color = colors[i])

Możesz użyć predefiniowanej „jakościowej mapy kolorów” w następujący sposób:

from matplotlib.cm import get_cmap

name = "Accent"
cmap = get_cmap(name)  # type: matplotlib.colors.ListedColormap
colors = cmap.colors  # type: list
axes.set_prop_cycle(color=colors)

Przetestowano na matplotlib 3.0.3. Zobacz https://github.com/matplotlib/matplotlib/issues/10840, aby dowiedzieć się, dlaczego nie możesz zadzwonić axes.set_prop_cycle(color=cmap).

Lista predefiniowanych jakościowych map kolorów jest dostępna pod adresem https://matplotlib.org/gallery/color/colormap_reference.html :

Możesz także zmienić domyślny cykl kolorów w matplotlibrcpliku. Jeśli nie wiesz, gdzie jest ten plik, wykonaj następujące czynności w Pythonie:

import matplotlib
matplotlib.matplotlib_fname()

Spowoduje to wyświetlenie ścieżki do aktualnie używanego pliku matplotlibrc. W tym pliku znajdziesz pośród wielu innych ustawień także to dla axes.color.cycle. Po prostu wprowadź żądaną sekwencję kolorów, a znajdziesz ją na każdej działce, którą stworzysz. Zauważ, że możesz także użyć wszystkich prawidłowych nazw kolorów HTML w matplotlib.

Jak zauważa odpowiedź Ciro , możesz użyć prop_cycledo ustawienia listy kolorów, przez które matplotlib będzie się przełączać. Ale ile kolorów? A co, jeśli chcesz użyć tego samego cyklu kolorów dla wielu wykresów, z różną liczbą linii?

Jedną z taktyk byłoby użycie formuły podobnej do tej z https://gamedev.stackexchange.com/a/46469/22397 , aby wygenerować nieskończoną sekwencję kolorów, w której każdy kolor stara się znacznie różnić od wszystkich, które go poprzedzały.

Niestety, prop_cyclenie przyjmie nieskończonych sekwencji - zawiesi się na zawsze, jeśli ją zdasz. Ale możemy wziąć, powiedzmy, pierwsze 1000 kolorów wygenerowanych z takiej sekwencji i ustawić je jako cykl kolorów. W ten sposób w przypadku działek z dowolną rozsądną liczbą linii należy uzyskać rozróżnialne kolory.

Przykład:

from matplotlib import pyplot as plt
from matplotlib.colors import hsv_to_rgb
from cycler import cycler

# 1000 distinct colors:
colors = [hsv_to_rgb([(i * 0.618033988749895) % 1.0, 1, 1])
          for i in range(1000)]
plt.rc('axes', prop_cycle=(cycler('color', colors)))

for i in range(20):
    plt.plot([1, 0], [i, i])

plt.show()

Wynik:

Teraz wszystkie kolory są inne - chociaż przyznaję, że trudno mi rozróżnić kilka z nich!