Merge pull request #170 from FarBo/axis_label_scale

Remove y-axis label from single panel
toolsforexperiments · Feb 5, 2021 · ac7141d · ac7141d
2 parents 7e6a1be + b34971c
commit ac7141d
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/plottr/plot/mpl.py b/plottr/plot/mpl.py
@@ -1034,11 +1034,12 @@ def _plot1dSinglepanel(self) -> None:
         axes = self._makeAxes(nAxes)
 
         if len(depvals) > 1:
-            ylbl = self.data.label(depnames[0])
-            phlbl: Optional[str] = f"Arg({depnames[0]})"
-        else:
             ylbl = None
-            phlbl = None
+            phlbl: Optional[str] = None
+        else:
+            ylbl = self.data.label(depnames[0])
+            phlbl = f"Arg({depnames[0]})"
+
 
         for yname, yvals in zip(depnames, depvals):
             # otherwise we sometimes raise ComplexWarning. This is basically just
@@ -1054,7 +1055,7 @@ def _plot1dSinglepanel(self) -> None:
                                 addLegend=(yname == depnames[-1]))
                     plot1dTrace(axes[1], xvals, np.asanyarray(yvals).imag,
                                 axLabels=(self.data.label(xname), ylbl),
-                                curveLabel=f"Im({yname})",
+                                curveLabel=f"Im({self.data.label(yname)})",
                                 addLegend=(yname == depnames[-1]))
                 elif self.complexRepresentation is ComplexRepresentation.magAndPhase:
                     plot1dTrace(axes[0], xvals, np.real(np.abs(yvals)),