簡體   English   中英

從 Bokeh 圖中返回 x、y 坐標

[英]Return x, y coordinates from Bokeh graph

我正在嘗試調整答案: Get selected data contained within box select tool in Bokeh

但得到:

NameError: name 'inds' is not defined 

選擇點后。

有誰知道發生了什么事?

我正在使用的代碼:

    import pandas as pd
    import numpy as np
    import matplotlib as mpl
    from bokeh.transform import factor_cmap, linear_cmap
    from bokeh.palettes import Spectral6
    from bokeh.plotting import figure, output_file, show
    from bokeh.models import CustomJS, ColumnDataSource
    from bokeh.io import output_notebook

    import bokeh.plotting as bpl
    import bokeh.models as bmo
    from bokeh.models.tools import *

    output_notebook()

    algorithm = 'Loc'
    metric = 'LengthOfStayDaysNBR'
    coordinate_df = pd.DataFrame({'x-{}'.format(str(algorithm)[:3]):np.random.rand(500)\
                                  ,'y-{}'.format(str(algorithm)[:3]):np.random.rand(500)\
                                 ,metric:np.random.rand(500)})


    TOOLS="pan,wheel_zoom,reset,hover,poly_select,lasso_select"


    p = figure(title = '{}'.format(str(algorithm)[:3]),tools=TOOLS)


    source = bpl.ColumnDataSource(coordinate_df)

    if metric == 'LengthOfStayDaysNBR':
        color_mapper = linear_cmap(metric,palette=Spectral6,low=coordinate_df[metric].min(),high=coordinate_df[metric].max())
    else:
        color_mapper = factor_cmap(metric,palette=Spectral6,factors=coordinate_df[metric].unique())



    p1 = figure(width=250, height = 250, title = '{}'.format(str(algorithm)[:3]),tools=TOOLS)

    p1.scatter('x-{}'.format(str(algorithm)[:3]),'y-{}'.format(str(algorithm)[:3]),fill_color= color_mapper, source = source)


    source.callback = CustomJS(args=dict(source=source), code="""
            var inds = cb_obj.get('selected')['1d'].indices;
            var d1 = cb_obj.get('data');
            console.log(d1)
            var kernel = IPython.notebook.kernel;
            IPython.notebook.kernel.execute("inds = " + inds);
            """
    )

    show(p1)


    # Run this after selecting
    for x,y in  zip([source.data['x-{}'.format(str(algorithm)[:3])][i] for i in inds],
        [source.data['y-{}'.format(str(algorithm)[:3])][i] for i in inds]):
        print(x,y)

提供一些背景知識:我正在嘗試從包含許多特征的較大數據框中獲取 x/y 坐標,選擇散點圖的子隊列,然后繪制(在條形圖中)特征的最大(20 左右)平均值對於子隊列的指定 x/y 坐標。 例如,假設您有 50 個特征(列)和 20 個數據點的子群組。 我希望將剩余的 48 個特征的平均值繪制成條形圖,條形圖的高度代表特征平均值。 知道怎么做嗎? – user123328 10 小時前
此外:如果能夠將實際索引/xy 坐標返回到某個對象(即數據框、np.array、列表或其他任何對象)中會很棒 – user123328 10 小時前

編輯:

我得到它(有點)與以下工作:

    def app(doc):

        x = df_algs['x-{}'.format(str(algorithm)[:3])]
        y = df_algs['y-{}'.format(str(algorithm)[:3])]
        # create the scatter plot


        if metric == 'LengthOfStayDaysNBR':
            color_mapper = linear_cmap(metric,palette=Spectral6,low=df_algs[metric].min(),high=df_algs[metric].max())
        else:
            color_mapper = factor_cmap(metric,palette=Spectral6,factors=df_algs[metric].unique())


        source = ColumnDataSource(dict(
            x = x
        ,   y = y
        ,   metric_ = df_algs[metric]))
        # create the scatter plot
        p = figure(tools=TOOLS, plot_width=600, plot_height=600, min_border=10, min_border_left=50,
                   toolbar_location="above", x_axis_location=None, y_axis_location=None,
                   title="Linked Histograms")
        p.select(BoxSelectTool).select_every_mousemove = False
        p.select(LassoSelectTool).select_every_mousemove = False

        r = p.scatter('x', 'y', source=source, fill_color = color_mapper, alpha=0.6)
    # 
        # create the horizontal histogram
        hhist, hedges = np.histogram(x, bins=20)
        hzeros = np.zeros(len(hedges)-1)
        hmax = max(hhist)*1.1

        LINE_ARGS = dict(color="#3A5785", line_color=None)

        ph = figure(toolbar_location=None, plot_width=p.plot_width, plot_height=200, x_range=p.x_range,
                    y_range=(-hmax, hmax), min_border=10, min_border_left=50, y_axis_location="right")
        ph.xgrid.grid_line_color = None
        ph.yaxis.major_label_orientation = np.pi/4
        ph.background_fill_color = "#fafafa"

        ph.quad(bottom=0, left=hedges[:-1], right=hedges[1:], top=hhist, color="white", line_color="#3A5785")
        hh1 = ph.quad(bottom=0, left=hedges[:-1], right=hedges[1:], top=hzeros, alpha=0.5, **LINE_ARGS)
        hh2 = ph.quad(bottom=0, left=hedges[:-1], right=hedges[1:], top=hzeros, alpha=0.1, **LINE_ARGS)

        # create the vertical histogram
        vhist, vedges = np.histogram(y, bins=20)
        vzeros = np.zeros(len(vedges)-1)
        vmax = max(vhist)*1.1

        pv = figure(toolbar_location=None, plot_width=200, plot_height=p.plot_height, x_range=(-vmax, vmax),
                    y_range=p.y_range, min_border=10, y_axis_location="right")
        pv.ygrid.grid_line_color = None
        pv.xaxis.major_label_orientation = np.pi/4
        pv.background_fill_color = "#fafafa"

        pv.quad(left=0, bottom=vedges[:-1], top=vedges[1:], right=vhist, color="white", line_color="#3A5785")
        vh1 = pv.quad(left=0, bottom=vedges[:-1], top=vedges[1:], right=vzeros, alpha=0.5, **LINE_ARGS)
        vh2 = pv.quad(left=0, bottom=vedges[:-1], top=vedges[1:], right=vzeros, alpha=0.1, **LINE_ARGS)

        layout = column(row(p, pv), row(ph, Spacer(width=200, height=200)))

        doc.add_root(layout)
        doc.title = "Selection Histogram"

        def update(attr, old, new):
            inds = np.array(new['1d']['indices'])
            if len(inds) == 0 or len(inds) == len(x):
                hhist1, hhist2 = hzeros, hzeros
                vhist1, vhist2 = vzeros, vzeros
            else:
                neg_inds = np.ones_like(x, dtype=np.bool)
                neg_inds[inds] = False
                hhist1, _ = np.histogram(x[inds], bins=hedges)
                vhist1, _ = np.histogram(y[inds], bins=vedges)
                hhist2, _ = np.histogram(x[neg_inds], bins=hedges)
                vhist2, _ = np.histogram(y[neg_inds], bins=vedges)

            hh1.data_source.data["top"]   =  hhist1
            hh2.data_source.data["top"]   = -hhist2
            vh1.data_source.data["right"] =  vhist1
            vh2.data_source.data["right"] = -vhist2

            df = df_algs.loc[df_algs.index.isin(inds)]
            df.drop(expected_metrics+metrics+['AgeNBR'],axis=1).mean().sort_values(ascending=False)[:25].plot(kind='Bar')
            plt.show()


        r.data_source.on_change('selected', update)

然而

        if metric == 'LengthOfStayDaysNBR':
            color_mapper = linear_cmap(metric,palette=Spectral6,low=df_algs[metric].min(),high=df_algs[metric].max())
        else:
            color_mapper = factor_cmap(metric,palette=Spectral6,factors=df_algs[metric].unique())

似乎打破了密碼。 嘗試對散點圖進行顏色編碼時,我得到“字形指的是不存在的列名稱:{}”。

我不會依賴調用kernel.executeCustomJS回調來執行此操作。 我認為這是一種非常脆弱的做事方式,當以任意順序重新執行單元格時,很容易造成混淆或不合理。 此外,為了將來的考慮,我不確定它是否會與下一代 JupyterLab 一起工作,因為kernel.execute可能不可用。

相反,我會嵌入一個真正的Bokeh 服務器應用程序 Bokeh 服務器的創建是為了通過定義和維護的協議在 Bokeh 的 JavaScript 和 Python 部分之間高效、穩健地同步數據,並根據任何一方的變化提供執行回調。

這是嵌入在筆記本中的應用程序的截屏視頻 gif,它根據中心圖中的選擇更新一對直方圖。 代碼單元在 gif 下方給出。

選擇直方圖筆記本應用程序


單元格 1

import numpy as np
from bokeh.io import output_notebook, show
from bokeh.layouts import row, column
from bokeh.models import BoxSelectTool, LassoSelectTool, Spacer
from bokeh.plotting import figure, curdoc

output_notebook()

單元格 2

x1 = np.random.normal(loc=5.0, size=400) * 100
y1 = np.random.normal(loc=10.0, size=400) * 10

x2 = np.random.normal(loc=5.0, size=800) * 50
y2 = np.random.normal(loc=5.0, size=800) * 10

x3 = np.random.normal(loc=55.0, size=200) * 10
y3 = np.random.normal(loc=4.0, size=200) * 10

x = np.concatenate((x1, x2, x3))
y = np.concatenate((y1, y2, y3))

TOOLS="pan,wheel_zoom,box_select,lasso_select,reset"

單元格 3

def app(doc):
    # create the scatter plot
    p = figure(tools=TOOLS, plot_width=600, plot_height=600, min_border=10, min_border_left=50,
               toolbar_location="above", x_axis_location=None, y_axis_location=None,
               title="Linked Histograms")
    p.background_fill_color = "#fafafa"
    p.select(BoxSelectTool).select_every_mousemove = False
    p.select(LassoSelectTool).select_every_mousemove = False

    r = p.scatter(x, y, size=3, color="#3A5785", alpha=0.6)

    # create the horizontal histogram
    hhist, hedges = np.histogram(x, bins=20)
    hzeros = np.zeros(len(hedges)-1)
    hmax = max(hhist)*1.1

    LINE_ARGS = dict(color="#3A5785", line_color=None)

    ph = figure(toolbar_location=None, plot_width=p.plot_width, plot_height=200, x_range=p.x_range,
                y_range=(-hmax, hmax), min_border=10, min_border_left=50, y_axis_location="right")
    ph.xgrid.grid_line_color = None
    ph.yaxis.major_label_orientation = np.pi/4
    ph.background_fill_color = "#fafafa"

    ph.quad(bottom=0, left=hedges[:-1], right=hedges[1:], top=hhist, color="white", line_color="#3A5785")
    hh1 = ph.quad(bottom=0, left=hedges[:-1], right=hedges[1:], top=hzeros, alpha=0.5, **LINE_ARGS)
    hh2 = ph.quad(bottom=0, left=hedges[:-1], right=hedges[1:], top=hzeros, alpha=0.1, **LINE_ARGS)

    # create the vertical histogram
    vhist, vedges = np.histogram(y, bins=20)
    vzeros = np.zeros(len(vedges)-1)
    vmax = max(vhist)*1.1

    pv = figure(toolbar_location=None, plot_width=200, plot_height=p.plot_height, x_range=(-vmax, vmax),
                y_range=p.y_range, min_border=10, y_axis_location="right")
    pv.ygrid.grid_line_color = None
    pv.xaxis.major_label_orientation = np.pi/4
    pv.background_fill_color = "#fafafa"

    pv.quad(left=0, bottom=vedges[:-1], top=vedges[1:], right=vhist, color="white", line_color="#3A5785")
    vh1 = pv.quad(left=0, bottom=vedges[:-1], top=vedges[1:], right=vzeros, alpha=0.5, **LINE_ARGS)
    vh2 = pv.quad(left=0, bottom=vedges[:-1], top=vedges[1:], right=vzeros, alpha=0.1, **LINE_ARGS)

    layout = column(row(p, pv), row(ph, Spacer(width=200, height=200)))

    doc.add_root(layout)
    doc.title = "Selection Histogram"

    def update(attr, old, new):
        inds = np.array(new['1d']['indices'])
        if len(inds) == 0 or len(inds) == len(x):
            hhist1, hhist2 = hzeros, hzeros
            vhist1, vhist2 = vzeros, vzeros
        else:
            neg_inds = np.ones_like(x, dtype=np.bool)
            neg_inds[inds] = False
            hhist1, _ = np.histogram(x[inds], bins=hedges)
            vhist1, _ = np.histogram(y[inds], bins=vedges)
            hhist2, _ = np.histogram(x[neg_inds], bins=hedges)
            vhist2, _ = np.histogram(y[neg_inds], bins=vedges)

        hh1.data_source.data["top"]   =  hhist1
        hh2.data_source.data["top"]   = -hhist2
        vh1.data_source.data["right"] =  vhist1
        vh2.data_source.data["right"] = -vhist2

    r.data_source.on_change('selected', update)

單元格 4

# set notebook_url appropriately
show(app, notebook_url="http://localhost:8889")

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM