这试图回答你所有的问题。下面的代码最多循环 7 种颜色。如果您需要更多,您应该创建一个更复杂的生成器,如另一个答案所示。
import numpy as np
from matplotlib import pyplot as plt
def get_color():
for item in ['r', 'g', 'b', 'c', 'm', 'y', 'k']:
yield item
x = 0.3 * np.array(range(40))
color = get_color()
for group in range(5):
# generates a collection of points
y = np.exp2(x + 0.5 * group)
# fit to a polynomial
z = np.polyfit(x, y, 6)
p = np.poly1d(z)
acolor = next(color)
plt.scatter(x, y, color=acolor, marker='o')
plt.plot(x, p(x), acolor + '-', label=str(group))
plt.legend()
plt.xlim((0, 15))
plt.show()
上面代码中的生成器对于示例来说有点矫枉过正,但它为更复杂的计算提供了结构。如果您只需要几种颜色,则可以使用简单的迭代器
>>> color = iter(list_of_colors)
>>> acolor = next(color)
如果您需要无休止地循环,您可以使用itertools.cycle
:
>>> from itertools import cycle
>>> color = cycle(['r', 'g', 'b', 'c', 'm', 'y', 'k'])
>>> next(color)
'r'
>>>
编辑:您有多种选择来获得 n 种不同的颜色。正如我之前指出的,您可以使用其他答案中指示的方法使用生成器。例如,用get_color
不同的生成器替换:
import colorsys
import numpy as np
from matplotlib import pyplot as plt
def get_color(color):
for hue in range(color):
hue = 1. * hue / color
col = [int(x) for x in colorsys.hsv_to_rgb(hue, 1.0, 230)]
yield "#{0:02x}{1:02x}{2:02x}".format(*col)
x = 0.3 * np.array(range(40))
color = get_color(15)
for group in range(15):
# generates a collection of points
y = np.exp2(x + 0.5 * group)
# fit to a polynomial
z = np.polyfit(x, y, 6)
p = np.poly1d(z)
acolor = next(color)
plt.scatter(x, y, color=acolor, marker='o')
plt.plot(x, p(x), color=acolor, linestyle='dashed', label=str(group))
plt.legend()
plt.xlim((0, 15))
plt.show()
你会得到 15 种不同的颜色。
然而,相似的颜色是连续的,不能提供良好的分辨率/对比度。您可以通过跳过色调值来增加对比度:
for hue in range(0, color*3, 3):
绘制多条线时的另一个问题是图例......