Python/Bokeh -如何通过使用Select、callback和CustomJS/js_on_change从dict中按列值筛选行来更改数据源

from bokeh.layouts import column
from bokeh.models import Select, CustomJS
from bokeh.plotting import curdoc
from bokeh.sampledata.autompg import data as df

columns = list(df.columns)
select = Select(title="Select a column:", options=columns)

callback = CustomJS(args=dict(source=source), code="""
    var data = source.data;
    var column = cb_obj.value;
    var selected_data = {};
    
    for (var key in data) {
        selected_data[key] = [];
        for (var i = 0; i < data[column].length; i++) {
            if (data[column][i] == column) {
                for (var key in data) {
                    selected_data[key].push(data[key][i]);
                }
            }
        }
    }
    
    source.data = selected_data;
""")

layout = column(select)
curdoc().add_root(layout)