Bokeh:使用CustomJS从checkboxgroup组更新散点图以过滤源

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

# 创建散点图
p = figure()
scatter = p.circle([1, 2, 3, 4, 5], [2, 5, 8, 2, 7])

# 创建checkboxgroup组件
checkbox_group = CheckboxGroup(labels=["A", "B", "C", "D", "E"], active=[0, 1, 2, 3, 4])

# 定义CustomJS回调函数
callback = CustomJS(args=dict(scatter=scatter, checkbox_group=checkbox_group), code="""
    // 获取checkboxgroup组件的选中状态
    var selected_indices = checkbox_group.active;
    
    // 创建新的数据源
    var new_data = {
        x: [],
        y: []
    };
    
    // 根据选中的索引更新数据源
    for (var i = 0; i < selected_indices.length; i++) {
        var index = selected_indices[i];
        new_data.x.push(scatter.data.x[index]);
        new_data.y.push(scatter.data.y[index]);
    }
    
    // 更新散点图的数据源
    scatter.data = new_data;
    scatter.change.emit();
""")

# 将回调函数绑定到checkboxgroup组件的active属性上
checkbox_group.js_on_change('active', callback)

# 将散点图和checkboxgroup组件放置在一起
layout = column(checkbox_group, p)

# 在Jupyter Notebook中显示结果
output_notebook()
show(layout)