plotlyで選択したデータの取得方法

Question

Plotlyに関して質問があります。

Plotlyは以下のような散布図を作り、Lasso selectや、Box selectで点を選択できるかと思います。

この選択したデータをcsvファイルなどで取得する方法はあるのでしょうか。

Answer 1

Plotlyはどのような環境で利用されていますか。
もしJupyter notebookやJupyterLab、Google Colaboratoryで利用されているのでしたら、FigureWidgetを使うことによって可能です。
ipywidgetsというPythonライブラリが必要ですので、なければインストールしておいてください。

以下のサンプルコードは、選択されたすべての点のx・y値を出力するとともに、ファイルに保存するコードです。

import numpy as np
import plotly.graph_objects as go

np.random.seed(1)
x = np.random.normal(2, 0.4, 400)
y = np.random.normal(2, 0.4, 400)

f = go.FigureWidget(data=go.Scatter(
    x=x,
    y=y,
    mode="markers",
    marker=dict(color="DarkOrange")
))

scatter = f.data[0]

def selection_fn(trace, points, selector):
    for x, y in zip(points.xs, points.ys):
        print(f"{x}, {y}")

    with open("coords_selected_points.csv", "w") as f:
        for x, y in zip(points.xs, points.ys):
            f.write(f"{x}, {y}\n")

scatter.on_selection(selection_fn)

f

Comments

[amumas]

誠にありがとうございます。少し勉強して自分で動かしてみます。またご連絡します。

[etherbeg]

すみません、上のコードはGoogle Colaboratoryではうまく表示されませんでした。自家製Widgetの使用には制限があるみたいです。