python - 如何阻止 Matplotlib 重复颜色？

Question

这个问题已经有答案了:

creating over 20 unique legend colors using matplotlib (4 个回答)

已关闭 4 年前。

我有这个代码:

plist = ['p5', 'p14', 'p23', 'p32', 'p41', 'p50', 'p59', 'p68', 'p77', 'p85', 'p95']


for pltcount in range(len(plist)):
    plt.plot(data1[pltcount], np.exp(data2)[pltcount], marker='o', label=str(plist[pltcount]))
plt.legend()
plt.show()

这是使用 plt.style.use('fivethirtyeight')让情节变得更好。我找到了手动分配颜色的示例。如果我希望它是自动的并且来自一些众所周知的调色板怎么办？

Answer 1

彩虹的颜色怎么样？这里的关键是使用ax.set_prop_cycle为每条线分配颜色。

NUM_COLORS = len(plist)

cm = plt.get_cmap('gist_rainbow')
fig = plt.figure()
ax = fig.add_subplot(111)
ax.set_prop_cycle('color', [cm(1.*i/NUM_COLORS) for i in range(NUM_COLORS)])
# Or,
# ax.set_prop_cycle(color=[cm(1.*i/NUM_COLORS) for i in range(NUM_COLORS)])
for i, p in enumerate(plist):
    ax.plot(data1[i], np.exp(data2)[i], marker='o', label=str(p))

plt.legend()
plt.show()

Borrowed from here 。其他可能的选择。