Changeset 2121 in lmdz_wrf

Timestamp:

Aug 31, 2018, 3:47:50 PM (7 years ago)

Author:

lfita

Message:

Adding 'fill' value to `running_mean'

File:

• trunk/tools/generic_tools.py (modified) (2 diffs)

trunk/tools/generic_tools.py

@@ -6335 +6335 @@
     return vecv
 
-def running_mean(values, run):
+def running_mean(values, run, fill=None):
     """ Function to compute a running mean of a series of values
       running_mean(vals, int)
       [vals]: series of values
       [run]: interval to use to compute the running mean 
-        NOTE: resultant size would be the same but with None except at [run/2,size(vals)-run/2]
+      [fill]: fill value to use
+        [value]: value to use to fill outside runmean values
+        'vals': using actual values instead
+        NOTE: resultant size would be the same but with [fill] except at [run/2,size(vals)-run/2]
       >>> running_mean(np.arange(100),10)
       [None None None None None 4.5 5.5 6.5 7.5 8.5 9.5 10.5 11.5 12.5 13.5 14.5
@@ -6362 +6365 @@
         runmean[ip] = np.mean(values[ip-run/2:ip+run/2])
 
-    runmean = np.where(runmean == fillValue, None, runmean)
+    if fill != 'vals':
+        runmean = np.where(runmean == fillValue, fill, runmean)
+    else:
+        runmean = np.where(runmean == fillValue, values, runmean)
+
 
     return runmean