mmo-modeling-results.py

trace_vars = [i.replace('coefficient_','') for i in dir(trace['posterior']) if i.startswith("coefficient_") or i=='intercept']
trace_coeff_vars = [i for i in dir(trace['posterior']) if i.startswith("coefficient_") or i=='intercept']
trace_alpha_vars = [i for i in dir(trace['posterior']) if i.startswith("alpha_")]
trace_gamma_vars = [i for i in dir(trace['posterior']) if i.startswith("gamma_")]

model_results = pd.DataFrame({
    "variable": trace_vars,
    "coefficient": [float(trace['posterior'][i].mean()) for i in trace_coeff_vars]
}).merge(
    pd.DataFrame({
      "variable": [i.replace('alpha_','') for i in trace_alpha_vars],
      "alpha": [float(trace['posterior'][i].mean()) for i in trace_alpha_vars],
      "gamma": [float(trace['posterior'][i].mean()) for i in trace_gamma_vars]
    }),
    'left'
)

model_results