widget sector violin

%23%20%2F%2F%2F%20script%0A%23%20requires-python%20%3D%20%22%3E%3D3.11%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%22numpy%22%2C%0A%23%20%20%20%20%20%22pandas%22%2C%0A%23%20%20%20%20%20%22pyarrow%22%2C%0A%23%20%20%20%20%20%22polars%22%2C%0A%23%20%20%20%20%20%22plotly%22%2C%0A%23%20%5D%0A%23%20%2F%2F%2F%0A%22%22%22Interactive%20widget%3A%20feature%20dropdown%20%E2%86%92%20sector-specific%20posterior%20violin%20plot.%0A%0AStandalone%20marimo%20notebook%20for%20WASM%20embedding.%20Loads%20the%20same%20data%20as%20the%0Afull%20blogpost%20but%20only%20renders%20the%20violin%20subplot%20for%20a%20user-selected%20feature.%0A%22%22%22%0A%0Aimport%20marimo%0A%0A__generated_with%20%3D%20%220.23.2%22%0Aapp%20%3D%20marimo.App(width%3D%22medium%22)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_()%3A%0A%20%20%20%20import%20marimo%20as%20mo%0A%0A%20%20%20%20return%20(mo%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20import%20io%0A%20%20%20%20import%20numpy%20as%20np%0A%20%20%20%20import%20pandas%20as%20pd%0A%20%20%20%20import%20plotly.graph_objects%20as%20go%0A%20%20%20%20import%20polars%20as%20pl%0A%0A%20%20%20%20BASE%20%3D%20mo.notebook_location()%20%2F%20%22public%22%0A%20%20%20%20DATA_DIR%20%3D%20BASE%20%2F%20%22data%22%0A%0A%20%20%20%20def%20_read_bytes(path)%3A%0A%20%20%20%20%20%20%20%20s%20%3D%20str(path)%0A%20%20%20%20%20%20%20%20if%20s.startswith((%22http%3A%2F%2F%22%2C%20%22https%3A%2F%2F%22))%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20import%20urllib.request%0A%20%20%20%20%20%20%20%20%20%20%20%20with%20urllib.request.urlopen(s)%20as%20r%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20r.read()%0A%20%20%20%20%20%20%20%20from%20pathlib%20import%20Path%20as%20_P%0A%20%20%20%20%20%20%20%20return%20_P(s).read_bytes()%0A%0A%20%20%20%20def%20read_parquet(path)%3A%0A%20%20%20%20%20%20%20%20return%20pl.from_pandas(pd.read_parquet(io.BytesIO(_read_bytes(path))))%0A%0A%20%20%20%20def%20read_npz(path)%3A%0A%20%20%20%20%20%20%20%20return%20np.load(io.BytesIO(_read_bytes(path))%2C%20allow_pickle%3DTrue)%0A%0A%20%20%20%20return%20DATA_DIR%2C%20go%2C%20np%2C%20pl%2C%20read_npz%2C%20read_parquet%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(DATA_DIR%2C%20np%2C%20pl%2C%20read_npz%2C%20read_parquet)%3A%0A%20%20%20%20%23%20Load%20diagnostics%20for%20mu_mean%20(group%20means)%0A%20%20%20%20_df%20%3D%20read_parquet(DATA_DIR%20%2F%20%22windows%22%20%2F%20%22summary_hier_mcmc.parquet%22)%0A%20%20%20%20_pred_date%20%3D%20_df%5B%22pred_date%22%5D.max()%0A%20%20%20%20_window%20%3D%20_df.filter(pl.col(%22pred_date%22)%20%3D%3D%20_pred_date)%0A%0A%20%20%20%20_bhat_rows%20%3D%20(%0A%20%20%20%20%20%20%20%20_window%0A%20%20%20%20%20%20%20%20.filter(pl.col(%22parameter%22).str.starts_with(%22beta_hat%5B%22))%0A%20%20%20%20%20%20%20%20.with_columns(%0A%20%20%20%20%20%20%20%20%20%20%20%20pl.col(%22parameter%22).str.extract(r%22beta_hat%5C%5B(.%2B)%5C%5D%22%2C%201).alias(%22feature%22)%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20)%0A%20%20%20%20_bhat%20%3D%20dict(zip(_bhat_rows%5B%22feature%22%5D.to_list()%2C%20_bhat_rows%5B%22mean%22%5D.to_list()))%0A%0A%20%20%20%20%23%20Load%20single-window%20draws%0A%20%20%20%20_loaded%20%3D%20read_npz(DATA_DIR%20%2F%20%22windows%22%20%2F%20%22single_window_data.npz%22)%0A%20%20%20%20single_window_data%20%3D%20%7B%0A%20%20%20%20%20%20%20%20%22beta_draws%22%3A%20%20_loaded%5B%22beta_draws%22%5D%2C%0A%20%20%20%20%20%20%20%20%22features%22%3A%20%20%20%20_loaded%5B%22features%22%5D.tolist()%2C%0A%20%20%20%20%20%20%20%20%22sectors%22%3A%20%20%20%20%20_loaded%5B%22sectors%22%5D.tolist()%2C%0A%20%20%20%20%20%20%20%20%22ols_coefs%22%3A%20%20%20_loaded%5B%22ols_coefs%22%5D%2C%0A%20%20%20%20%20%20%20%20%22lasso_coefs%22%3A%20_loaded%5B%22lasso_coefs%22%5D%2C%0A%20%20%20%20%20%20%20%20%22mu_mean%22%3A%20%20%20%20%20np.array(%5B_bhat%5Bf%5D%20for%20f%20in%20_loaded%5B%22features%22%5D.tolist()%5D)%2C%0A%20%20%20%20%7D%0A%20%20%20%20return%20(single_window_data%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo%2C%20single_window_data)%3A%0A%20%20%20%20feature_dropdown%20%3D%20mo.ui.dropdown(%0A%20%20%20%20%20%20%20%20options%3Dsingle_window_data%5B%22features%22%5D%2C%0A%20%20%20%20%20%20%20%20value%3Dsingle_window_data%5B%22features%22%5D%5B0%5D%2C%0A%20%20%20%20%20%20%20%20label%3D%22**Characteristic**%22%2C%0A%20%20%20%20)%0A%20%20%20%20feature_dropdown%0A%20%20%20%20return%20(feature_dropdown%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(feature_dropdown%2C%20go%2C%20mo%2C%20single_window_data)%3A%0A%20%20%20%20_feat%20%3D%20feature_dropdown.value%0A%20%20%20%20_sectors%20%3D%20single_window_data%5B%22sectors%22%5D%0A%20%20%20%20_mu_val%20%3D%20float(single_window_data%5B%22mu_mean%22%5D%5Bsingle_window_data%5B%22features%22%5D.index(_feat)%5D)%0A%0A%20%20%20%20_feat_idx%20%3D%20single_window_data%5B%22features%22%5D.index(_feat)%0A%20%20%20%20_ols_val%20%3D%20float(single_window_data%5B%22ols_coefs%22%5D%5B_feat_idx%5D)%0A%20%20%20%20_lasso_val%20%3D%20single_window_data%5B%22lasso_coefs%22%5D%5B_feat_idx%5D%0A%0A%20%20%20%20_fig%20%3D%20go.Figure()%0A%0A%20%20%20%20for%20_s%20in%20_sectors%3A%0A%20%20%20%20%20%20%20%20_s_idx%20%3D%20single_window_data%5B%22sectors%22%5D.index(_s)%0A%20%20%20%20%20%20%20%20_draws%20%3D%20single_window_data%5B%22beta_draws%22%5D%5B%3A%2C%20_s_idx%2C%20_feat_idx%5D%0A%20%20%20%20%20%20%20%20_fig.add_trace(go.Violin(%0A%20%20%20%20%20%20%20%20%20%20%20%20x%3D%5B_s%5D%20*%20len(_draws)%2C%20y%3D_draws.tolist()%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20name%3D_s%2C%20showlegend%3DFalse%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20box_visible%3DFalse%2C%20meanline_visible%3DTrue%2C%20points%3DFalse%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20fillcolor%3D%22rgba(76%2C%20114%2C%20176%2C%200.35)%22%2C%20line_color%3D%22%234C72B0%22%2C%0A%20%20%20%20%20%20%20%20))%0A%0A%20%20%20%20_fig.add_trace(go.Scatter(%0A%20%20%20%20%20%20%20%20x%3D_sectors%2C%20y%3D%5B_mu_val%5D%20*%20len(_sectors)%2C%0A%20%20%20%20%20%20%20%20mode%3D%22lines%22%2C%20name%3Df%22Hierarchical%20(Group%20mean)%20%3D%20%7B_mu_val%3A.4f%7D%22%2C%0A%20%20%20%20%20%20%20%20line%3Ddict(color%3D%22%2355A868%22%2C%20dash%3D%22dot%22%2C%20width%3D2)%2C%0A%20%20%20%20))%0A%20%20%20%20_fig.add_trace(go.Scatter(%0A%20%20%20%20%20%20%20%20x%3D_sectors%2C%20y%3D%5B_ols_val%5D%20*%20len(_sectors)%2C%0A%20%20%20%20%20%20%20%20mode%3D%22lines%22%2C%20name%3Df%22OLS%20(pooled)%20%3D%20%7B_ols_val%3A.4f%7D%22%2C%0A%20%20%20%20%20%20%20%20line%3Ddict(color%3D%22%23d62728%22%2C%20dash%3D%22dash%22%2C%20width%3D2)%2C%0A%20%20%20%20))%0A%20%20%20%20_fig.add_trace(go.Scatter(%0A%20%20%20%20%20%20%20%20x%3D_sectors%2C%20y%3D%5Bfloat(_lasso_val)%5D%20*%20len(_sectors)%2C%0A%20%20%20%20%20%20%20%20mode%3D%22lines%22%2C%20name%3Df%22LASSO%20(%CE%B1%3D5%C3%9710%E2%81%BB%E2%81%B4)%20%3D%20%7B_lasso_val%3A.4f%7D%22%2C%0A%20%20%20%20%20%20%20%20line%3Ddict(color%3D%22%23e377c2%22%2C%20dash%3D%22dashdot%22%2C%20width%3D2)%2C%0A%20%20%20%20))%0A%0A%20%20%20%20_fig.update_layout(%0A%20%20%20%20%20%20%20%20title%3Df%22Sector%20posterior%20distributions%20%E2%80%94%20%7B_feat%7D%22%2C%0A%20%20%20%20%20%20%20%20xaxis_title%3D%22Sector%22%2C%0A%20%20%20%20%20%20%20%20yaxis_title%3Df%22%CE%B2%20(%7B_feat%7D)%22%2C%0A%20%20%20%20%20%20%20%20height%3D500%2C%0A%20%20%20%20%20%20%20%20legend%3Ddict(orientation%3D%22h%22%2C%20yanchor%3D%22top%22%2C%20y%3D-0.32%2C%20xanchor%3D%22center%22%2C%20x%3D0.5)%2C%0A%20%20%20%20%20%20%20%20margin%3Ddict(t%3D60%2C%20b%3D120%2C%20l%3D30%2C%20r%3D20)%2C%0A%20%20%20%20%20%20%20%20violinmode%3D%22overlay%22%2C%0A%20%20%20%20)%0A%20%20%20%20mo.vstack(%5Bfeature_dropdown%2C%20mo.ui.plotly(_fig)%5D)%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