New _Rprofs.stepstr property

Author: amorison

stagpy/_step.py

@@ -460,6 +460,11 @@ def __getitem__(self, name: str) -> Rprof:
 
         return Rprof(rprof, rad, meta)
 
+    @property
+    def stepstr(self) -> str:
+        """String representation of the parent :class:`Step`."""
+        return str(self.step.istep)
+
     @crop
     def centers(self) -> ndarray:
         """Radial position of cell centers."""

stagpy/rprof.py

@@ -21,10 +21,7 @@ def plot_rprofs(rprofs: _Rprofs, names: str):
         names: profile names separated by ``-`` (figures), ``.`` (subplots) and
             ``,`` (same subplot).
     """
-    try:
-        stepstr = rprofs.steps.stepstr
-    except AttributeError:
-        stepstr = str(rprofs.step.istep)
+    stepstr = rprofs.stepstr
     sdat = rprofs.step.sdat
 
     for vfig in _helpers.list_of_vars(names):

stagpy/stagyydata.py

@@ -334,6 +334,11 @@ def __getitem__(self, name: str) -> Rprof:
         self._cached_data[name] = Rprof(rprof, rad, meta)
         return self._cached_data[name]
 
+    @property
+    def stepstr(self) -> str:
+        """String representation of steps indices."""
+        return self.steps.stepstr
+
 
 class _Steps:
     """Collections of time steps.