web-dev-qa-db-fra.com

Comment définir le cycle de couleur par défaut pour toutes les sous-parcelles avec matplotlib?

Comment puis-je définir un jeu de couleurs par défaut pour les tracés réalisés avec matplotlib? Je peux définir une carte de couleurs particulière comme celle-ci

import numpy as np
import matplotlib.pyplot as plt

fig=plt.figure(i)
ax=plt.gca()
colormap = plt.get_cmap('jet')
ax.set_color_cycle([colormap(k) for k in np.linspace(0, 1, 10)])

mais existe-t-il un moyen de définir le même jeu de couleurs pour tous les tracés, y compris les sous-tracés?

pythonmatplotlibcolors
56
Katt

Sûr! Soit spécifiez axes.color_cycle dans ton .matplotlibrc file ou définissez-le lors de l'exécution à l'aide de matplotlib.rcParams ou matplotlib.rc.

À titre d'exemple de ce dernier:

import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np

# Set the default color cycle
mpl.rcParams['axes.color_cycle'] = ['r', 'k', 'c']

# Alternately, we could use rc:
# mpl.rc('axes', color_cycle=['r','k','c'])

x = np.linspace(0, 20, 100)

fig, axes = plt.subplots(nrows=2)

for i in range(10):
    axes[0].plot(x, i * (x - 10)**2)

for i in range(10):
    axes[1].plot(x, i * np.cos(x))

plt.show()

enter image description here

68
Joe Kington

À partir de matplotlib 1.5, mpl.rcParams ['axes.color_cycle'] est déconseillé. Vous devez utiliser axes.prop_cycle:

import matplotlib as mpl
mpl.rcParams['axes.prop_cycle'] = mpl.cycler(color=["r", "#e94cdc", "0.7"])
41
volodymyr

Dans la version 2.1.0, ce qui suit fonctionne pour moi, en utilisant set_prop_cycle et module cycler

from cycler import cycler
custom_cycler = (cycler(color=['r','b','m','g']))
ax.set_prop_cycle(custom_cycler)

vous pouvez ajouter un attribut de ligne supplémentaire

custom_cycler = (cycler(color=['r','b','m','g']) + cycler(lw=[1,1,1,2]))

'ax' vient de ax = plt.axes () ou de tout générateur d'axes

0
Joonho Park