使用 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)

现在我们将生成一个包含两个 Axes 的图形，一个在另一个之上。在第一个轴上，我们将使用默认的 cycler 进行绘图。在第二个轴上，我们将使用 matplotlib.axes.Axes.set_prop_cycle() 设置 prop_cycle，这只会为这个 matplotlib.axes.Axes 实例设置 prop_cycle。我们将使用第二个 cycler，它结合了颜色 cycler 和线宽 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': '-.'}