【发布时间】：2018-03-27 00:58:55
【问题描述】：
我正在尝试调整来自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、列表或其他任何东西——10 小时前的 user123328
编辑：
我得到它（有点）使用以下内容：
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())
似乎破坏了代码。尝试对散点图进行颜色编码时，我得到“字形引用不存在的列名：{}”。
【问题讨论】：
标签： python callback bokeh lasso-regression