¿Cómo elegir un nuevo color para cada línea trazada dentro de una figura en matplotlib?

Me gustaría NO especificar un color para cada línea trazada y que cada línea tenga un color distinto. Pero si corro:

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

plt.show()

entonces obtengo esta salida:

Si observa la imagen de arriba, puede ver que matplotlib intenta elegir colores para cada línea que son diferentes, pero eventualmente reutiliza los colores: las diez líneas superiores usan los mismos colores que las diez inferiores. Solo quiero evitar que repita colores ya usados ​​Y / O alimentarlo con una lista de colores para usar.

matplotlib 1.5+

Puede usar axes.set_prop_cycle( ejemplo ).

matplotlib 1.0-1.4

Puede usar axes.set_color_cycle( ejemplo ).

matplotlib 0.x

Puede utilizar Axes.set_default_color_cycle.

Normalmente uso el segundo de estos:

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)

Ejemplo de 2:

prop_cycle

color_cyclefue obsoleto en 1.5 a favor de esta generalización: 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()

También se muestra en el ejemplo (ahora mal nombrado): http://matplotlib.org/1.5.1/examples/color/color_cycle_demo.html mencionado en: https://stackoverflow.com/a/4971431/895245

Probado en matplotlib 1.5.1.

No sé si puede cambiar el color automáticamente, pero podría aprovechar su bucle para generar diferentes colores:

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

En este caso, los colores variarán del negro al 100% verde, pero puedes afinarlo si quieres.

Consulte los documentos de matplotlib plot () y busque el colorargumento de palabra clave.

Si desea alimentar una lista de colores, solo asegúrese de tener una lista lo suficientemente grande y luego use el índice del ciclo para seleccionar el color

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

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

Puede utilizar un "mapa de colores cualitativo" predefinido como este:

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)

Probado en matplotlib 3.0.3. Consulte https://github.com/matplotlib/matplotlib/issues/10840 para obtener información sobre por qué no puede llamar axes.set_prop_cycle(color=cmap).

Una lista de mapas de color cualitativos predefinidos está disponible en https://matplotlib.org/gallery/color/colormap_reference.html :

También puede cambiar el ciclo de color predeterminado en su matplotlibrcarchivo. Si no sabe dónde está ese archivo, haga lo siguiente en Python:

import matplotlib
matplotlib.matplotlib_fname()

Esto le mostrará la ruta a su archivo matplotlibrc actualmente utilizado. En ese archivo encontrará, entre muchas otras configuraciones, también la de axes.color.cycle. Simplemente coloque la secuencia de colores deseada y la encontrará en cada trama que haga. Tenga en cuenta que también puede utilizar todos los nombres de colores HTML válidos en matplotlib.

Como señala la respuesta de Ciro , puede usarprop_cycle para establecer una lista de colores para que matplotlib recorra el ciclo. ¿Pero cuántos colores? ¿Qué sucede si desea utilizar el mismo ciclo de color para muchos gráficos, con diferentes números de líneas?

Una táctica sería utilizar una fórmula como la de https://gamedev.stackexchange.com/a/46469/22397 , para generar una secuencia infinita de colores donde cada color intenta ser significativamente diferente de todos los que lo precedieron.

Desafortunadamente, prop_cycleno aceptará secuencias infinitas, se bloqueará para siempre si le pasa una. Pero podemos tomar, digamos, los primeros 1000 colores generados a partir de dicha secuencia y establecerlo como ciclo de color. De esa manera, para las parcelas con cualquier número de líneas, debería obtener colores distinguibles.

Ejemplo:

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

Salida:

Ahora, todos los colores son diferentes, ¡aunque admito que me cuesta distinguir algunos de ellos!