Use rprofs_averaged instead of computing by hand

Author: amorison

stagpy/plates.py

@@ -574,16 +574,9 @@ def set_of_vars(arg_plot):
 
 def main_plates(sdat):
     """Plot several plates information."""
-    # calculating averaged horizontal surface velocity
-    # needed for redimensionalisation
-    rlast = sdat.snaps[-1].rprofs
-    nprof = 0
-    uprof_averaged = np.zeros_like(rlast['vhrms'].values)
-    for step in sdat.walk.filter(rprofs=True):
-        uprof_averaged += step.rprofs['vhrms'].values
-        nprof += 1
-    uprof_averaged /= nprof
-    radius = rlast['r'].values
+    # averaged horizontal surface velocity needed for redimensionalisation
+    uprof_averaged, radius, _ = sdat.walk.filter(rprofs=True)\
+        .rprofs_averaged['vhrms']
     if sdat.par['boundaries']['air_layer']:
         dsa = sdat.par['boundaries']['air_thickness']
         isurf = np.argmin(abs(radius - radius[-1] + dsa))