使用 Bokeh 绘图时，如何自动循环显示调色板？

Question

我想使用循环加载和/或修改数据并使用 Bokeh 在循环内绘制结果（我熟悉 Matplotlib's axes.color_cycle）。这是一个简单的例子

import numpy as np
from bokeh.plotting import figure, output_file, show
output_file('bokeh_cycle_colors.html')

p = figure(width=400, height=400)
x = np.linspace(0, 10)

for m in xrange(10):
    y = m * x
    p.line(x, y, legend='m = {}'.format(m))

p.legend.location='top_left'
show(p)

生成此图

如何使颜色循环而不编码颜色列表和模数运算以在颜色数量用完时重复？

在 GitHub 上有一些与此相关的讨论，问题 351 和 2201，但不清楚如何使其工作。我在documentation 中搜索cycle color 时得到的四个点击实际上并没有在页面的任何位置包含单词cycle。

Answer 1

获取颜色列表并使用itertools自行循环可能是最简单的：

import numpy as np
from bokeh.plotting import figure, output_file, show

# select a palette
from bokeh.palettes import Dark2_5 as palette
# itertools handles the cycling
import itertools  

output_file('bokeh_cycle_colors.html')

p = figure(width=400, height=400)
x = np.linspace(0, 10)

# create a color iterator
colors = itertools.cycle(palette)    

for m, color in zip(range(10), colors):
    y = m * x
    p.line(x, y, legend='m = {}'.format(m), color=color)

p.legend.location='top_left'
show(p)

Answer 2

您可以定义一个简单的生成器，为您循环颜色。

在 python 3 中：

from bokeh.palettes import Category10
import itertools

def color_gen():
    yield from itertools.cycle(Category10[10])
color = color_gen()

或在python 2（或3）中：

from bokeh.palettes import Category10
import itertools

def color_gen():
    for c in itertools.cycle(Category10[10]):
        yield c
color = color_gen()

当您需要新颜色时，请执行以下操作：

p.line(x, y1, line_width=2, color=color)
p.line(x, y2, line_width=2, color=color)

这是上面的例子：

p = figure(width=400, height=400)
x = np.linspace(0, 10)

for m, c in zip(range(10), color):
    y = m * x
    p.line(x, y, legend='m = {}'.format(m), color=c)

p.legend.location='top_left'
show(p)

Answer 3

两个小改动将使之前的答案适用于 Python 3。

• 已更改：for m, color in zip(range(10), colors):

• 之前：for m, color in itertools.izip(xrange(10), colors):

Answer 4

在 Python > 3.7 中，您可以执行以下操作：

from bokeh.palettes import Category10_10
       
color = Category10_10.__iter__()

p.line(x, y1, line_width=2, color=next(color))

这将循环遍历列表的每个元素，直到您每次使用 next() 时用尽。

python 中的每个序列类型都可以返回一个迭代器对象。