mayutils 1.2.40__tar.gz → 1.2.41__tar.gz

{mayutils-1.2.40 → mayutils-1.2.41}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mayutils
-Version: 1.2.40
+Version: 1.2.41
 Summary: Utilities for Python from Mayuran Visakan
 Author: Mayuran Visakan
 Author-email: Mayuran Visakan <mayuran.k.v@gmail.com>

{mayutils-1.2.40 → mayutils-1.2.41}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mayutils"
-version = "1.2.40"
+version = "1.2.41"
 description = "Utilities for Python from Mayuran Visakan"
 authors = [
     { name = "Mayuran Visakan", email = "mayuran.k.v@gmail.com" }

{mayutils-1.2.40 → mayutils-1.2.41}/src/mayutils/visualisation/graphs/plotly/charts.py

@@ -1325,57 +1325,45 @@ class Plot(go.Figure):
         y_max: Optional[float] = None,
         y_padding: float = 0.05,
     ) -> Self:
-        for yaxis in [prop for prop in self.layout if prop.startswith("yaxis")]:
-            yaxis_suffix = yaxis.removeprefix("yaxis")
-
-            trace_limits = np.asarray(
+        yaxes = sorted([prop for prop in self.layout if prop.startswith("yaxis")])
+        trace_limits_full = {
+            yaxis: np.asarray(
                 [
                     (
                         np.nanmin(visible_y)
-                        if not (np.isnan(visible_y := trace.y[visible_mask])).all()
+                        if not np.isnan(visible_y := trace.y[visible_mask]).all()
                         and not visible_y.shape == (0,)
                         else np.nan,
                         np.nanmax(visible_y)
-                        if not (np.isnan(visible_y)).all()
-                        and not visible_y.shape == (0,)
+                        if not np.isnan(visible_y).all() and not visible_y.shape == (0,)
                         else np.nan,
                     )
                     for trace in self.data
-                    if (trace.visible is None or trace.visible is True)
+                    if trace.visible in [None, True]
                     and isinstance(trace.y, np.ndarray)
-                    and trace.yaxis == f"y{yaxis_suffix}"
+                    and getattr(
+                        self.layout, trace.yaxis.replace("y", "yaxis")
+                    ).matches.replace("y", "yaxis")
+                    == yaxis
                     and (
                         visible_mask := (trace.x < self.layout.xaxis.range[1])  # type: ignore
                         & (trace.x > self.layout.xaxis.range[0])  # type: ignore
                     ).any()
                 ]
             )
+            for yaxis in yaxes
+        }
 
-            traces_min = (
-                np.nanmin(trace_limits) if not np.isnan(trace_limits).all() else None
+        trace_limits = {
+            f"{yaxis}_range": (
+                np.nanmin(limits[:, 0]) if not np.isnan(limits[:, 0]).all() else None,
+                np.nanmax(limits[:, 1]) if not np.isnan(limits[:, 1]).all() else None,
             )
-            traces_max = (
-                np.nanmax(trace_limits) if not np.isnan(trace_limits).all() else None
-            )
-
-            if traces_min is not None and traces_max is not None:
-                span = traces_max - traces_min
-
-                y_range = (
-                    (traces_min - span * y_padding).clip(min=y_min),
-                    (traces_max + span * y_padding).clip(max=y_max),
-                )
-            else:
-                y_range = (
-                    traces_min * (1 - y_padding) if traces_min is not None else None,
-                    traces_max * (1 + y_padding) if traces_max is not None else None,
-                )
+            for yaxis, limits in trace_limits_full.items()
+            if len(limits) > 0
+        }
 
-            self.update_layout(
-                {
-                    f"yaxis{yaxis_suffix}_range": y_range,
-                }
-            )
+        self.update_layout(trace_limits)
 
         return self