matplotlib 多种颜色用于绘图

Question

我有很多图想要在一个图表中展示。我刚开始学习使用matplotlib，但找不到一个好的方法来生成很多不同的颜色 :( 也许可以在HSV颜色空间中循环，保持饱和度和明度最大？

我在想类似这样的代码

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()