chartselect

%23%20%2F%2F%2F%20script%0A%23%20requires-python%20%3D%20%22%3E%3D3.10%22%0A%23%20dependencies%20%3D%20%5B%0A%23%20%20%20%20%20%22marimo%22%2C%0A%23%20%20%20%20%20%22matplotlib%22%2C%0A%23%20%20%20%20%20%22numpy%22%2C%0A%23%20%20%20%20%20%22wigglystuff%3D%3D0.2.18%22%2C%0A%23%20%5D%0A%23%20%2F%2F%2F%0A%0Aimport%20marimo%0A%0A__generated_with%20%3D%20%220.19.7%22%0Aapp%20%3D%20marimo.App()%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20import%20marimo%20as%20mo%0A%20%20%20%20import%20matplotlib.pyplot%20as%20plt%0A%20%20%20%20import%20numpy%20as%20np%0A%20%20%20%20from%20wigglystuff%20import%20ChartSelect%0A%20%20%20%20return%20ChartSelect%2C%20mo%2C%20np%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(%22%22%22%0A%20%20%20%20%23%20ChartSelect%20Demo%0A%0A%20%20%20%20Interactive%20region%20selection%20for%20matplotlib%20charts.%0A%0A%20%20%20%20-%20Use%20**Box**%20or%20**Lasso**%20mode%20to%20draw%20a%20selection%0A%20%20%20%20-%20Click%20and%20drag%20inside%20the%20selection%20to%20move%20it%0A%20%20%20%20-%20Click%20outside%20the%20selection%20to%20clear%20it%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(np)%3A%0A%20%20%20%20np.random.seed(42)%0A%20%20%20%20x_data%20%3D%20np.random.randn(300)%0A%20%20%20%20y_data%20%3D%20np.random.randn(300)%0A%20%20%20%20return%20x_data%2C%20y_data%0A%0A%0A%40app.cell%0Adef%20_(ChartSelect%2C%20x_data%2C%20y_data)%3A%0A%20%20%20%20def%20draw_chart(ax%2C%20widget)%3A%0A%20%20%20%20%20%20%20%20if%20widget.has_selection%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20mask%20%3D%20widget.get_mask(x_data%2C%20y_data)%0A%20%20%20%20%20%20%20%20%20%20%20%20ax.scatter(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20x_data%5B~mask%5D%2C%20y_data%5B~mask%5D%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20alpha%3D0.3%2C%20color%3D%22gray%22%2C%20s%3D20%0A%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20%20%20%20%20ax.scatter(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20x_data%5Bmask%5D%2C%20y_data%5Bmask%5D%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20alpha%3D0.8%2C%20color%3D%22%23ef4444%22%2C%20s%3D40%0A%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20%20%20%20%20ax.set_title(f%22Selected%3A%20%7Bmask.sum()%7D%20%2F%20%7Blen(x_data)%7D%20points%22)%0A%20%20%20%20%20%20%20%20else%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20ax.scatter(x_data%2C%20y_data%2C%20alpha%3D0.6%2C%20color%3D%22%233b82f6%22)%0A%20%20%20%20%20%20%20%20%20%20%20%20ax.set_title(%22Draw%20a%20selection%20to%20highlight%20points%22)%0A%20%20%20%20%20%20%20%20ax.set_xlabel(%22X%22)%0A%20%20%20%20%20%20%20%20ax.set_ylabel(%22Y%22)%0A%20%20%20%20%20%20%20%20ax.grid(True%2C%20alpha%3D0.3)%0A%0A%20%20%20%20select%20%3D%20ChartSelect.from_callback(%0A%20%20%20%20%20%20%20%20draw_fn%3Ddraw_chart%2C%0A%20%20%20%20%20%20%20%20x_bounds%3D(-3.5%2C%203.5)%2C%0A%20%20%20%20%20%20%20%20y_bounds%3D(-3.5%2C%203.5)%2C%0A%20%20%20%20%20%20%20%20figsize%3D(5%2C%205)%2C%0A%20%20%20%20%20%20%20%20mode%3D%22box%22%2C%0A%20%20%20%20%20%20%20%20selection_color%3D%22%23ef4444%22%2C%0A%20%20%20%20)%0A%20%20%20%20return%20(select%2C)%0A%0A%0A%40app.cell%0Adef%20_(mo%2C%20select)%3A%0A%20%20%20%20widget%20%3D%20mo.ui.anywidget(select)%0A%20%20%20%20return%20(widget%2C)%0A%0A%0A%40app.cell%0Adef%20_(widget)%3A%0A%20%20%20%20widget%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(mo%2C%20widget%2C%20x_data%2C%20y_data)%3A%0A%20%20%20%20_msg%20%3D%20%22No%20selection%20yet%22%0A%20%20%20%20if%20widget.has_selection%3A%0A%20%20%20%20%20%20%20%20_mask%20%3D%20widget.get_mask(x_data%2C%20y_data)%0A%20%20%20%20%20%20%20%20_n%20%3D%20_mask.sum()%0A%20%20%20%20%20%20%20%20if%20_n%20%3E%200%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20_msg%20%3D%20f%22Selected%20%7B_n%7D%20points%22%0A%20%20%20%20%20%20%20%20else%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20_msg%20%3D%20%22Selection%20contains%200%20points%22%0A%20%20%20%20mo.callout(_msg)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(%22%22%22%0A%20%20%20%20%23%23%20Single-Mode%20Widgets%0A%0A%20%20%20%20You%20can%20restrict%20a%20widget%20to%20only%20one%20selection%20mode%20by%20passing%20%60modes%3D%5B%22box%22%5D%60%20or%20%60modes%3D%5B%22lasso%22%5D%60.%0A%20%20%20%20When%20only%20one%20mode%20is%20available%2C%20the%20mode%20buttons%20are%20hidden.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(ChartSelect%2C%20np)%3A%0A%20%20%20%20np.random.seed(123)%0A%20%20%20%20_x%20%3D%20np.random.randn(150)%0A%20%20%20%20_y%20%3D%20np.random.randn(150)%0A%0A%20%20%20%20def%20_draw_box(ax%2C%20widget)%3A%0A%20%20%20%20%20%20%20%20ax.scatter(_x%2C%20_y%2C%20alpha%3D0.6%2C%20color%3D%22%233b82f6%22%2C%20s%3D30)%0A%20%20%20%20%20%20%20%20if%20widget.has_selection%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20mask%20%3D%20widget.get_mask(_x%2C%20_y)%0A%20%20%20%20%20%20%20%20%20%20%20%20ax.scatter(_x%5Bmask%5D%2C%20_y%5Bmask%5D%2C%20alpha%3D0.9%2C%20color%3D%22%2322c55e%22%2C%20s%3D50)%0A%20%20%20%20%20%20%20%20ax.set_title(%22Box%20only%22)%0A%20%20%20%20%20%20%20%20ax.grid(True%2C%20alpha%3D0.3)%0A%0A%20%20%20%20def%20_draw_lasso(ax%2C%20widget)%3A%0A%20%20%20%20%20%20%20%20ax.scatter(_x%2C%20_y%2C%20alpha%3D0.6%2C%20color%3D%22%233b82f6%22%2C%20s%3D30)%0A%20%20%20%20%20%20%20%20if%20widget.has_selection%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20mask%20%3D%20widget.get_mask(_x%2C%20_y)%0A%20%20%20%20%20%20%20%20%20%20%20%20ax.scatter(_x%5Bmask%5D%2C%20_y%5Bmask%5D%2C%20alpha%3D0.9%2C%20color%3D%22%23f59e0b%22%2C%20s%3D50)%0A%20%20%20%20%20%20%20%20ax.set_title(%22Lasso%20only%22)%0A%20%20%20%20%20%20%20%20ax.grid(True%2C%20alpha%3D0.3)%0A%0A%20%20%20%20box_only%20%3D%20ChartSelect.from_callback(%0A%20%20%20%20%20%20%20%20draw_fn%3D_draw_box%2C%0A%20%20%20%20%20%20%20%20x_bounds%3D(-3%2C%203)%2C%0A%20%20%20%20%20%20%20%20y_bounds%3D(-3%2C%203)%2C%0A%20%20%20%20%20%20%20%20figsize%3D(4%2C%204)%2C%0A%20%20%20%20%20%20%20%20mode%3D%22box%22%2C%0A%20%20%20%20%20%20%20%20modes%3D%5B%22box%22%5D%2C%0A%20%20%20%20%20%20%20%20selection_color%3D%22%2322c55e%22%2C%0A%20%20%20%20)%0A%0A%20%20%20%20lasso_only%20%3D%20ChartSelect.from_callback(%0A%20%20%20%20%20%20%20%20draw_fn%3D_draw_lasso%2C%0A%20%20%20%20%20%20%20%20x_bounds%3D(-3%2C%203)%2C%0A%20%20%20%20%20%20%20%20y_bounds%3D(-3%2C%203)%2C%0A%20%20%20%20%20%20%20%20figsize%3D(4%2C%204)%2C%0A%20%20%20%20%20%20%20%20mode%3D%22lasso%22%2C%0A%20%20%20%20%20%20%20%20modes%3D%5B%22lasso%22%5D%2C%0A%20%20%20%20%20%20%20%20selection_color%3D%22%23f59e0b%22%2C%0A%20%20%20%20)%0A%20%20%20%20return%20box_only%2C%20lasso_only%0A%0A%0A%40app.cell%0Adef%20_(box_only%2C%20lasso_only%2C%20mo)%3A%0A%20%20%20%20_box_widget%20%3D%20mo.ui.anywidget(box_only)%0A%20%20%20%20_lasso_widget%20%3D%20mo.ui.anywidget(lasso_only)%0A%20%20%20%20mo.hstack(%5B_box_widget%2C%20_lasso_widget%5D%2C%20justify%3D%22start%22%2C%20gap%3D1)%0A%20%20%20%20return%0A%0A%0Aif%20__name__%20%3D%3D%20%22__main__%22%3A%0A%20%20%20%20app.run()%0A