matplotlib/examples/color/color_cycle_demo.py at master · feisuo/matplotlib · GitHub

Repository navigation

"""

Demo of custom property-cycle settings to control colors and such

for multi-line plots.

This example demonstrates two different APIs:

1. Setting the default rc-parameter specifying the property cycle.

This affects all subsequent axes (but not axes already created).

2. Setting the property cycle for a specific axes. This only

affects a single axes.

"""

from cycler import cycler

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)

# Create array with shifted-sine curve along each column

yy = np.transpose([np.sin(x + phi) for phi in offsets])

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

plt.rc('axes', prop_cycle=(cycler('color', ['r', 'g', 'b', 'y']) +

cycler('linestyle', ['-', '--', ':', '-.'])))

fig, (ax0, ax1) = plt.subplots(nrows=2)

ax0.plot(yy)

ax0.set_title('Set default color cycle to rgby')

ax1.set_prop_cycle(cycler('color', ['c', 'm', 'y', 'k']) +

cycler('lw', [1, 2, 3, 4]))

ax1.plot(yy)

ax1.set_title('Set axes color cycle to cmyk')

# Tweak spacing between subplots to prevent labels from overlapping

plt.subplots_adjust(hspace=0.3)

plt.show()