Changing colour scheme of python matplotlib python plots [duplicate]

Question

I want to set a colour scheme for my python plots, so that they don't repeat the same colour like they are for A and H in the top plot shown below. (sorry if its difficult to see).

The code I'm using is simply,

ax1.plot(sections,A,label='A',linewidth=2) ax1.plot(sections,B,label='B',linewidth=2) ax1.plot(sections,C,label='C',linewidth=2) ax1.plot(sections,D,label='D',linewidth=2) ax1.plot(sections,E,label='E',linewidth=2) ax1.plot(sections,F,label='F',linewidth=2) ax1.plot(sections,G,label='G',linewidth=2) ax1.plot(sections,H,label='H',linewidth=2)

What's the best way to set the colour scheme? Is using the colourmap function? Thanks for any help!

Answer 1

You can use some colormaps to color your lines, example (here I use the 'jet' colormap):

>>> from matplotlib import cm
>>> for i, val in enumerate(cm.jet(linspace(0,1,10))): #different lines
    plt.plot(arange(10), arange(10)+i, color=val, linestyle='-')

To do this in your existing code with minimal changes, just add (And don't forget to change 10 to the number of plots that you have.):

for L, C in zip([item for item in ax.get_children() if isinstance(item, matplotlib.lines.Line2D)], cm.jet(linspace(0,1,10))):
    L.set_color(C)   

Answer 2

You can change the color map for a specific axes by using the set_color_cycle() method.

This is a simplified version of the color cycle demo in the matplotlib docs:

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 2 * np.pi)
offsets = np.linspace(0, 2*np.pi, 4, endpoint=False)
yy = np.transpose([np.sin(x + phi) for phi in offsets])

plt.rc('lines', linewidth=4)

ax = plt.gca()

# Set color cycle for this axes 
ax.set_color_cycle(['c', 'm', 'y', 'k'])

ax.plot(yy)

plt.show()

The other option is to change the global default color map. You can do this by either creating a matplotlibrc file and setting axes.color_cyle there, or by chaning the running configuration at runtime:

import matplotlib
matplotlib.rcParams['axes.color_cycle'] = ['r', 'g', 'b']

You can also use HTML hex notation for specifying colors:

ax.set_color_cycle(['#FF0000', '#00FF00', '#0000FF'])

For more information on how to specify colors, see the documentation of the `color``module.

The matplotlib.cm module is also useful, for example for registering your own color maps.