return dict of contour values with labels, including best value

Author: bmaranville

refl1d/webview/server/profile_uncertainty.py

@@ -239,12 +239,13 @@ def _draw_contours(
         col=col,
         secondary_y=secondary_y,
     )
-    q = _plot_quantiles(
+    named_contours = _plot_quantiles(
         zp, fp, contours, color, alpha=None, fig=fig, row=row, col=col, legendgroup=legendgroup, secondary_y=secondary_y
     )
+    named_contours["best"] = fp[0]
     # Plot the best
     # axes.plot(zp, fp[0], '-', label=label, color=dark(color))
-    return q
+    return named_contours
 
 
 def _profile_labels(fig: "go.Figure", row: Optional[int] = None, col: Optional[int] = None, magnetic: bool = False):
@@ -366,9 +367,12 @@ def _plot_quantiles(
     """
     _, q = form_quantiles(y, contours)
 
+    output = {}
     if alpha is None:
         alpha = 2.0 / (len(q) + 1)
-    for lo, hi in q:
+    for contour, (lo, hi) in zip(contours, q):
+        output[f"{contour} percent lower"] = lo
+        output[f"{contour} percent upper"] = hi
         fig.add_scattergl(
             x=x,
             y=lo,
@@ -396,4 +400,4 @@ def _plot_quantiles(
             legendgroup=legendgroup,
             secondary_y=secondary_y,
         )
-    return q
+    return output