用Javascript过滤Bokeh LabelSet

from bokeh.plotting import figure, show
from bokeh.models import ColumnDataSource, LabelSet, CustomJS

# 创建数据源
source = ColumnDataSource(data=dict(
    x=[1, 2, 3, 4, 5],
    y=[1, 2, 3, 4, 5],
    labels=['A', 'B', 'C', 'D', 'E']
))

# 创建图表
p = figure(plot_width=400, plot_height=400)
p.circle('x', 'y', size=10, source=source)

# 创建LabelSet组件
labels = LabelSet(x='x', y='y', text='labels', source=source)

# 创建Javascript过滤逻辑
filter_code = """
    var labels = labels.data;
    var visible_labels = [];
    for (var i = 0; i < labels.length; i++) {
        if (labels[i] != 'B') {  // 根据特定条件筛选标签
            visible_labels.push(labels[i]);
        }
    }
    labels.text = visible_labels;
"""

# 创建CustomJS回调函数
callback = CustomJS(args=dict(labels=labels.data_source), code=filter_code)

# 将过滤逻辑应用于LabelSet组件
labels.js_on_change('text', callback)

# 将LabelSet组件添加到图表中
p.add_layout(labels)

# 显示图表
show(p)