如何在 matplotlib 图形上循环显示颜色和线条样式?

How to cycle through both colours and linestyles on a matplotlib figure?

提问人:CiaranWelsh 提问时间:5/16/2018 更新时间:5/17/2018 访问量:3320

问:

我有一堆图表,每个图表上大约有 15 到 20 条线。我想循环使用颜色和线条样式来获得许多独特的线条。如果我使用 6 种颜色和 4 种线型,则应该有 20 条唯一的线条,但下面的代码只产生 6 条。我做错了什么?

这是一个假数据集

import matplotlib.pyplot as plt
from cycler import cycler
import numpy
import seaborn
seaborn.set_style('white')

x = range(10)
ys = []
for i in range(20):
    ys.append(numpy.random.uniform(1, 10, size=10)*i)

这是我可以从其他帖子中收集到的:

plt.rc('axes', prop_cycle=(cycler('color', ['r', 'g', 'b', 'y', 'c', 'k']) +
                           cycler('linestyle', ['-', '--', ':', '-.', '-', '--'])))

注意:我在线型循环仪中使用了重复项,否则我们会得到 ValueError。

和绘图

plt.figure()
for i in range(20):
    plt.plot(x, ys[i], label=i)
plt.legend(loc=(1, 0.1))
plt.show()

enter image description here

python matplotlib

评论


答:

8赞 CodeZero 5/16/2018 #1

您可能想将两个循环仪相乘:

plt.rc('axes', prop_cycle=(cycler('color', ['r', 'g', 'b', 'y', 'c', 'k']) *
                           cycler('linestyle', ['-', '--', ':', '-.'])))

如上所述,加法同时循环颜色和线型,而乘法混合所有属性。

在这种情况下,您不必使用重复的线型,因为颜色和线型不必具有相等的周期长度。

6赞 WoofDoggy 5/17/2018 #2

正如 @datasailor 已经提到的,您应该将两个周期相乘:

cycler_op1 = cycler('color', ['r', 'g', 'b', 'y', 'c', 'k']) \
            * cycler('linestyle', ['-', '--', ':', '-.', '-', '--'])

cycler_op2 = cycler('linestyle', ['-', '--', ':', '-.', '-', '--']) \
            * cycler('color', ['r', 'g', 'b', 'y', 'c', 'k'])

rc('axes', prop_cycle = cycler_op1 ) # or cycler_op2

请注意,乘法不是交换的,你会得到不同的结果。基本上,在第一种情况下,颜色是固定的,线条样式会发生变化。在第二种情况下,线型是固定的,颜色会发生变化。总共 6x6 = 36 种可能性。

cycler_op1

cycler_op2

对于如此多的曲线,您可以尝试更多的颜色和线条样式。

ls_cycler = cycler('linestyle',
                    [(0,()), # solid 
                     (0, (1, 10)), # loosely dotted
                     (0, (1, 5)), # dotted
                     (0, (1, 1)), # densely dotted
                     (0, (5, 10)), # loosely dashed
                     (0, (5, 5)), # dashed
                     (0, (5, 1)), # densely dashed
                     (0, (3, 10, 1, 10)), # loosely dashdotted
                     (0, (3, 5, 1, 5)), # dashdotted
                     (0, (3, 1, 1, 1)), # densely dashdotted
                     (0, (3, 10, 1, 10, 1, 10)), # loosely dashdotdotted
                     (0, (3, 5, 1, 5, 1, 5)), # dashdotdotted
                     (0, (3, 1, 1, 1, 1, 1))] # densely dashdotdotted
                  )

color_cycler = cycler('color', [plt.get_cmap('jet')(i/13) for i in range(13)] )

new_cycler = color_cycler + ls_cycler

结果如下所示:

new_cycler