使用colormaps和matplotlib循环仪

Question

我想使用matplotlib循环仪和palettable的颜色。

cycler看起来像这样：

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

如何使用从palettable获取的颜色映射替换上面的颜色列表？

import palettable
cmap = palettable.colorbrewer.diverging.PRGn_11.mpl_colormap

对于答案，使用palettable并不重要，但了解如何使用colormap非常重要。

Answer 1

cycler需要将一个iterable分配给'colors'。

这是一种可以生成一种方式的方法：

[plt.get_cmap('jet')(1. * i/n) for i in range(n)]

从原来的例子来看：

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

x = [1,2,3,4]
for i in range(4):
    plt.plot([_ + i for _ in x])

来自'jet'colormap的修改后的列表：

n = 4 # Number of colors
new_colors = [plt.get_cmap('jet')(1. * i/n) for i in range(n)]

plt.rc('axes', 
       prop_cycle=(cycler('color', new_colors) + 
                   cycler('linestyle', ['-', '--', ':', '-.'])))

for i in range(4):
    plt.plot([_ + i for _ in x])