"""

====================

Petroff style sheets

====================

This example demonstrates the "petroffN" styles, which implement the 6-, 8- and

10-color sequences developed by Matthew A. Petroff [1]_ for accessible data

visualization. The styles balance aesthetics with accessibility considerations,

making them suitable for various types of plots while ensuring readability and

distinction between data series.

.. [1] https://arxiv.org/abs/2107.02270

"""

import matplotlib.pyplot as plt

import numpy as np

def colored_lines_example(ax):

t = np.linspace(-10, 10, 100)

nb_colors = len(plt.rcParams['axes.prop_cycle'])

shifts = np.linspace(-5, 5, nb_colors)

amplitudes = np.linspace(1, 1.5, nb_colors)

for t0, a in zip(shifts, amplitudes):

y = a / (1 + np.exp(-(t - t0)))

line, = ax.plot(t, y, '-')

point_indices = np.linspace(0, len(t) - 1, 20, dtype=int)

ax.plot(t[point_indices], y[point_indices], 'o', color=line.get_color())

ax.set_xlim(-10, 10)

def image_and_patch_example(ax):

ax.imshow(np.random.random(size=(20, 20)), interpolation='none')

c = plt.Circle((5, 5), radius=5, label='patch')

ax.add_patch(c)

fig = plt.figure(figsize=(6.4, 9.6), layout='compressed')

sfigs = fig.subfigures(nrows=3)

for style, sfig in zip(['petroff6', 'petroff8', 'petroff10'], sfigs):

sfig.suptitle(f"'{style}' style sheet")

with plt.style.context(style):

ax1, ax2 = sfig.subplots(ncols=2)

colored_lines_example(ax1)

image_and_patch_example(ax2)

plt.show()