matplotlib之使用cycler设置属性

matplotlib色彩（colors）之图表数据系列默认配色（默认色彩循环）和matplotlib之pyplot模块plot函数基础三（线条外观：Line2d关键字参数**kwargs）说了matplotlib绘图的默认外观属性与rcParams相关，其中绘图中的多个系列的颜色是根据默认色彩循环设定的。
能不能通过修改rcParams修改绘图的属性呢？

默认属性循环

输出默认的rcParams["axes.prop_cycle"]参数可知，对应的值为cycler对象，键为color，即默认只设置了颜色的属性循环。

import matplotlib.pyplot as plt
print(plt.rcParams["axes.prop_cycle"])

cycler('color', ['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728', '#9467bd', 
'#8c564b', '#e377c2', '#7f7f7f', '#bcbd22', '#17becf'])

修改默认属性循环

matplotlib提供了三种种修改属性循环的方法：

• 全局修改：
  • 通过plt.rc()修改。例如plt.rc('axes', prop_cycle=default_cycler)。
  • 修改matplotlibrc文件。例如axes.prop_cycle : default_cycler
• 指定子图修改：通过子图实例的set_prop_cycle()修改，例如ax1.set_prop_cycle(default_cycler)。

案例

全局修改色彩循环，指定修改子图ax1的线宽。

from cycler import cycler
import matplotlib.pyplot as plt

# 全局修改rcParams["axes.prop_cycle"]，对所有子图生效
default_cycler = (cycler(color=['r', 'g', 'b', 'y']) )
plt.rc('axes', prop_cycle=default_cycler)

fig, (ax0, ax1) = plt.subplots(nrows=2)
for i in range(5):
    ax0.plot([i,i])
print(plt.rcParams["axes.prop_cycle"])

# 指定修改ax1，只对子图ax1生效
custom_cycler = (cycler(lw=[1, 2, 3, 4]))
ax1.set_prop_cycle(custom_cycler)
for i in range(5):
    ax1.plot([i,i])
print(plt.rcParams["axes.prop_cycle"])
print([next(ax1._get_lines.prop_cycler) for i in range(5)])

plt.show()

案例解析

第二个子图中，多个线条并没有按全局修改的色彩循环交替更改颜色，为什么呢？
根据set_prop_cycle方法可知，一旦使用set_prop_cycle方法，就不再从rcParams['axes.prop_cycle']取值，而plot方法中，如果没有设置颜色的相关参数，就会线条的配色设置为C0，C0即rcParams['axes.prop_cycle']中的第一个元素，这里即红色，所以第二个子图中所有线条的颜色都为红色。

# axes._base模块
def set_prop_cycle(self, *args, **kwargs):
    # Can't do `args == (None,)` as that crashes cycler.
    if not (args or kwargs) or (len(args) == 1 and args[0] is None):
        prop_cycler = mpl.rcParams['axes.prop_cycle']
    else:
        prop_cycler = cycler(*args, **kwargs)

    self.prop_cycler = itertools.cycle(prop_cycler)
    # This should make a copy
    self._prop_keys = prop_cycler.keys

# axes._axes模块plot方法
if ('color' in kwargs or 'color' in fmt_style_kwargs):
    base_style = {}
    if 'color' in kwargs:
        base_style['color'] = kwargs.pop('color')
else:
    base_style = next(self._get_lines.prop_cycler)

# axes._axes模块plot方法
if 'color' not in base_style:
    base_style['color'] = 'C0'
if ecolor is None:
    ecolor = base_style['color']

结论

使用rcParams['axes.prop_cycle']全局修改默认属性循环比较简单，关键是熟悉cycler对象的用法。可参考matplotlib之属性组合包（cycler）
使用子图实例的set_prop_cycle()方法修改指定子图的默认属性循环必须设置色彩循环，要不然多个系列配色相同。