在 JupyterLab 中提取 plotly.express 选择

Question

我想从 plotly.express 图中所做的选择中提取索引或掩码。 该图是在 JupyterLab 中创建的。

import plotly.express as px
df = px.data.iris()
fig = px.scatter(df x="sepal_width", y="sepal_length", color="species")
fig.show()

此图显示未触及的图形。

此图显示了任意选择。 从这个选择中，我想提取索引列表或 boolean 掩码，或任何允许从原始 DataFrame 中提取选择的内容。

似乎有一些属性/函数可以帮助解决这个问题，例如 fig.data[0].selectedpoints。 我无法使用它们。

plotly 是版本：'4.14.3'

Answer 1

据我所知，没有办法获得用户选择的范围。 您在问题中指出的特征selectedpoints可供绘图员用来突出显示特定范围。 它可以作为创建者的场景而不是用户的选择。 我已使用此页面中的信息自定义此功能。

import  plotly.graph_objects as go
import numpy as np

df = px.data.iris()

fig = go.Figure()
fig.add_trace(go.Scatter(x=df['sepal_width'],
                         y=df['sepal_length'],
                         mode='markers',
                         marker=dict(color='rgba(0, 45, 240)', size=10)))

fig.update_layout(width=600, 
              height=550, 
              autosize=False,
              xaxis=dict(zeroline=False),
              hovermode='closest')

fig.show()

inds = [15+k for k in range(30)]
fig.data[0].update(selectedpoints=inds,
                   selected=dict(marker=dict(color='red')),#color of selected points
                   unselected=dict(marker=dict(color='rgb(200,200, 200)',#color of unselected pts
                                   opacity=0.9)));
fig.show()