Bokeh 在 Python Bokeh 绘图中选择图元

在本文中，我们将介绍如何在 Bokeh 中选择图元。Bokeh 是一个用于构建互动性可视化应用程序的 Python 库。它提供了丰富的绘图和数据可视化功能，使得用户可以轻松地创建各种类型的图表和图形。

Bokeh 提供了几种选择图元的方法，每种方法都有其特定的用途和适用范围。在下面的示例中，我们将说明如何使用 Bokeh 的选择功能来突出显示特定的图元。

阅读更多：Bokeh 教程

选择整个图

要选择整个图，可以使用 Bokeh 提供的 SelectTool 工具。这个工具可以让用户在图上拖动一个矩形框来选择图中的图元。当用户拖动矩形框时，所有图元都会被突出显示。

下面的代码示例演示了如何创建一个具有选择功能的 Bokeh 图。我们首先创建一个包含一些数据的数据源，并使用 square 方法在图上绘制正方形。然后，我们将 SelectTool 添加到图中，并将其设置为默认活动工具。最后，我们呈现图形并显示结果。

from bokeh.models import SelectTool
from bokeh.plotting import figure, show

# 创建数据源
x = [1, 2, 3, 4, 5]
y = [6, 7, 2, 4, 5]

# 创建绘图对象
p = figure(tools="pan,wheel_zoom,reset")

# 绘制正方形
p.square(x, y, size=10)

# 添加选择工具
p.add_tools(SelectTool())

# 设置选择工具为默认活动工具
p.toolbar.active_drag = p.select_one(SelectTool)

# 呈现图形并显示结果
show(p)

运行上述代码，将会显示一个具有选择功能的 Bokeh 图。当用户拖动鼠标时，图中的正方形将被突出显示。

选择特定图元

除了选择整个图之外，Bokeh 还允许用户选择特定的图元。这在需要对某些图元进行特殊处理或分析时非常有用。

要选择特定的图元，可以使用 SelectEvent 类中的 geometries 属性。该属性包含了用户选择的图元的几何信息。下面的示例展示了如何通过选择图形来突出显示图中的某些点。

from bokeh.events import SelectionGeometry
from bokeh.models import Circle, CustomJS
from bokeh.plotting import figure, show

# 创建数据源
x = [1, 2, 3, 4, 5]
y = [6, 7, 2, 4, 5]

# 创建绘图对象
p = figure(tools="box_select,reset")

# 绘制圆
circles = p.circle(x, y, size=10)

# 创建选择回调函数
select_cb = CustomJS(args=dict(circles=circles), code="""
    const selected = cb_obj.geometries;
    const { data_source } = circles;

    // 重置所有点的颜色
    data_source.data.fill = 'blue';

    // 根据选择的点修改颜色为红色
    for (let i = 0; i < selected.length; i++) {
        const index = selected[i].index;
        data_source.data.fill[index] = 'red';
    }

    // 触发数据源更新
    data_source.change.emit();
""")

# 添加选择回调函数
p.js_on_event(SelectionGeometry, select_cb)

# 呈现图形并显示结果
show(p)

上述代码中，我们首先创建了一组带有坐标的数据点，并使用 circle 方法在图上绘制这些点。然后，我们创建了一个自定义的 JavaScript 回调函数，该函数会在选择事件触发时执行。在此回调函数中，我们首先将所有点的颜色重置为蓝色，然后根据选择的点将颜色修改为红色。最后，我们将此回调函数与图的选择事件关联起来，并呈现图形。

当用户在图上选择点时，选择的点将以红色突出显示，而其他点将保持为蓝色。