Bokeh:如何使用复选框进行交互式绘图来更新图例？

from bokeh.plotting import figure, show
from bokeh.models import CheckboxGroup
from bokeh.layouts import column

# 创建一个绘图对象
p = figure()

# 创建复选框组件
checkbox_group = CheckboxGroup(labels=["数据1", "数据2", "数据3"], active=[0, 1, 2])

# 定义数据
data1 = [1, 2, 3, 4, 5]
data2 = [2, 4, 6, 8, 10]
data3 = [3, 6, 9, 12, 15]

# 绘制初始图形
line1 = p.line(range(len(data1)), data1, line_color="red", legend_label="数据1")
line2 = p.line(range(len(data2)), data2, line_color="green", legend_label="数据2")
line3 = p.line(range(len(data3)), data3, line_color="blue", legend_label="数据3")

# 定义复选框的回调函数
def update_plot(attr, old, new):
    # 根据复选框的选择更新图例
    line1.visible = 0 in checkbox_group.active
    line2.visible = 1 in checkbox_group.active
    line3.visible = 2 in checkbox_group.active

# 将复选框的回调函数与复选框组件关联
checkbox_group.on_change("active", update_plot)

# 创建布局并显示
layout = column(checkbox_group, p)
show(layout)