使用 cycler 进行样式设置

自定义属性循环设置的演示，用于控制多线图的颜色和其他样式属性。

注意

有关 cycler API 的更完整文档，请参见 此处。

此示例演示了两种不同的 API

1. 设置指定默认属性循环的 rc 参数。这会影响所有后续的 Axes（但不影响已创建的 Axes）。

2. 为单个 Axes 对设置属性循环。

from cycler import cycler

import matplotlib.pyplot as plt
import numpy as np

首先，我们将生成一些示例数据，在本例中，是四个偏移正弦曲线。

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

现在 yy 的形状为

print(yy.shape)

(50, 4)

因此 yy[:, i] 将为您提供第 i 个偏移正弦曲线。让我们使用 matplotlib.pyplot.rc() 设置默认的 prop_cycle。我们将通过将两个 cycler 相加 (+) 来组合颜色循环器和线型循环器。有关组合不同循环器的更多信息，请参见本教程的底部。

default_cycler = (cycler(color=['r', 'g', 'b', 'y']) +
                  cycler(linestyle=['-', '--', ':', '-.']))

plt.rc('lines', linewidth=4)
plt.rc('axes', prop_cycle=default_cycler)

现在我们将生成一个包含两个坐标轴的图形，一个在另一个上面。在第一个坐标轴上，我们将使用默认的循环器进行绘图。在第二个坐标轴上，我们将使用 prop_cycle 设置 matplotlib.axes.Axes.set_prop_cycle()，这只会为这个 matplotlib.axes.Axes 实例设置 prop_cycle。我们将使用第二个 cycler，它组合了颜色循环器和线宽循环器。

custom_cycler = (cycler(color=['c', 'm', 'y', 'k']) +
                 cycler(lw=[1, 2, 3, 4]))

fig, (ax0, ax1) = plt.subplots(nrows=2)
ax0.plot(yy)
ax0.set_title('Set default color cycle to rgby')
ax1.set_prop_cycle(custom_cycler)
ax1.plot(yy)
ax1.set_title('Set axes color cycle to cmyk')

# Add a bit more space between the two plots.
fig.subplots_adjust(hspace=0.3)
plt.show()

在 matplotlibrc 文件或样式文件中设置 prop_cycle

请记住，可以在您的 matplotlibrc 文件或样式文件 (style.mplstyle) 中的 axes.prop_cycle 下设置自定义循环器。

axes.prop_cycle : cycler(color='bgrcmyk')

循环遍历多个属性

您可以添加循环器

from cycler import cycler
cc = (cycler(color=list('rgb')) +
      cycler(linestyle=['-', '--', '-.']))
for d in cc:
    print(d)

结果为

{'color': 'r', 'linestyle': '-'}
{'color': 'g', 'linestyle': '--'}
{'color': 'b', 'linestyle': '-.'}

您可以将循环器相乘

from cycler import cycler
cc = (cycler(color=list('rgb')) *
      cycler(linestyle=['-', '--', '-.']))
for d in cc:
    print(d)

结果为

{'color': 'r', 'linestyle': '-'}
{'color': 'r', 'linestyle': '--'}
{'color': 'r', 'linestyle': '-.'}
{'color': 'g', 'linestyle': '-'}
{'color': 'g', 'linestyle': '--'}
{'color': 'g', 'linestyle': '-.'}
{'color': 'b', 'linestyle': '-'}
{'color': 'b', 'linestyle': '--'}
{'color': 'b', 'linestyle': '-.'}