matplotlib большой набор цветов для сюжетов

Question

У меня много графиков, которые я хочу построить на одном графике. Я только начал с matplotlib и не могу найти хороший способ для генерации множества различимых цветов :( Может быть, цикл на HSV с SV максимален?

Я думаю о чем-то вроде

args=[]
for i,(x,y) in enumerate(data):
    args.extend([x,y,hsv(i)])
plot(*args)

Есть предложения? :)

Answer 1

Я думаю, у вас правильная идея, за исключением того, что цвета будут более различимы, если вы передадите цветовую карту hsv чисел, разбросанных по диапазону (0,1):

hsv = plt.get_cmap('hsv')
hsv(float(i)/(len(data)-1))

или, используя NumPy:

colors = hsv(np.linspace(0, 1.0, len(kinds)))

Например:

import datetime as DT
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.dates as mdates
import scipy.interpolate as interpolate

dates = [DT.date(year, 9, 1) for year in range(2003, 2009)]
t = list(map(mdates.date2num, dates))
jec = (100, 70, 125, 150, 300, 250)
plt.plot(dates, jec, 'k.', markersize = 20)
new_t = np.linspace(min(t), max(t), 80)
new_dates = map(mdates.num2date, new_t)
kinds = ('cubic', 'quadratic', 'slinear', 'nearest', 'linear', 'zero', 4, 5)
cmap = plt.get_cmap('jet')
colors = cmap(np.linspace(0, 1.0, len(kinds)))
for kind, color in zip(kinds, colors):
    new_jec = interpolate.interp1d(t, jec, kind=kind)(new_t)
    plt.plot(new_t, new_jec, '-', label=str(kind), color=color)
plt.legend(loc = 'best')
plt.show()