Matplotlib颜色和线周期问题

Question

我正在编写一个包含适合我需要的所有参数的小类，但是当我定义要完成的3个周期时，我遇到了一个问题。我所做的是：定义3种类型的循环（cycle0，cycle1，cycle2），然后在包含几个rcParams的字典中设置'axes.prop_cycle'，3循环定义如下：

cycle0 =  plt.cycler("color", colors)         
cycle1 = plt.cycler("color", colors) + plt.cycler("linestyle", linestyles)
cycle2 = plt.cycler("color", colors) * plt.cycler("linestyle", linestyles)

当我选择使用循环0时出现问题：

'axes.prop_cycle': cycle0, 

并且颜色的集合（我选择）与linestyle有不同的长度但是使用cycle0我不会使用linestyle！我该如何处理这个问题？到目前为止，当我必须使用cycle0时，我必须注释掉cycle1，当我想要使用它时注释掉cycle1

编辑错误是这一个：

ValueError: Can only add equal length cycles, not 6 and 8

但在myparams定义中我使用了

'axes.prop_cycle': cycle0,

编辑最小可编辑示例：

import numpy as np
import matplotlib.pyplot as plt

def _plot(*args,**kwargs):

    linestyles  = [(0, ()),(0, (1, 1)),(0, (5, 10)),(0, (5, 5)),(0, (5, 1)),(0, (3, 10, 1, 10)),(0, (3, 5, 1, 5)),(0, (3, 5, 1, 5, 1, 5)) ]
    colors      = ['#3B4CC0','#7C9FF9','#C0D4F5','#F2CBB7','#EE8568','#B40426']

    variable = [*args]
    if len(variable) % 3 != 0:
      print('Error variable must be coupled (even number)')
      raise AttributeError('you must give 2 array x,y followed by string label for each plot')

    parameters = {**kwargs} 

    cycle0 =  plt.cycler("color", colors) 
    cycle1 = plt.cycler("color", colors) + plt.cycler("linestyle", linestyles)
    cycle2 = plt.cycler("color", colors) * plt.cycler("linestyle", linestyles)


    myparams = {
        'axes.prop_cycle': cycle0  
     }

    plt.rcParams.update(myparams)
    fig,ax = plt.subplots(1,figsize=(7,4.2))

    for i in range(0,len(variable)-1,3):
            plt.plot(variable[i],variable[i+1], linewidth=3, label= variable[i+2])
    plt.grid(linestyle='--', dashes=(5,7))
    plt.show() 



def main():

    x  = np.linspace(0,2*np.pi,100)
    y1 = np.sin(x) 
    y2 = np.sin(2*x)
    y3 = np.sin(4*x)

    _plot(x,y1,'sin(x)',x,y2,'sin(2x)',x,y3,'sin(3x)')


if __name__ == '__main__' :
    main()

如果相反你评论cycle1这是有效的！

编辑也许我不能很好地解释，这个代码（cycle1）是一个循环，我使用不同的颜色方案..我报告的那个只是一个例子说：

此外，如果我在rcParams列表中使用cycle0（这里我已经报告），只需将颜色方案从蓝色变为红色。但在我的课堂上有很多颜色方案，还有很多线条样式....出现的参数

myparams = {
        ......
        'axes.prop_cycle': cycle0  
        ......
         }

从字典中给出（看起来是_plot的定义）我希望fugure out是一种方法让这个循环1作为我的'axes.prop_cycle'的选项，这样我可以切换注释或取消注释这行代码（cycle1 ... 。）

Answer 1

问题出在cycle1，而不是cycle0。您可以删除该行，因为它从未在代码中使用过。 否则，您需要确保具有相同长度的循环器。

例如，您可以在颜色循环中添加两种颜色

colors = ['#3B4CC0','#7C9FF9','#C0D4F5','#F2CBB7','#EE8568','#B40426', "red", "limegreen"]

这样，线条和颜色都有8个元素。