使用 cycler 设置样式

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

备注

有关 cycler API 的更完整文档,请 here .

此示例演示了两个不同的 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 的图形,一个在另一个之上. 在第一个轴上,我们将使用默认的循环器进行绘制. 在第二个轴上,我们将使用 matplotlib.axes.Axes.set_prop_cycle() 设置 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': '-.'}