Changeset 669 in lmdz_wrf

Timestamp:

Jan 15, 2016, 2:14:07 PM (9 years ago)

Author:

lfita

Message:

Fixing weighted mean

File:

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

trunk/tools/nc_var_tools.py

@@ -17274 +17274 @@
                   "variable '" + varname2 + "' !!"
                 quit(-1)
-            iovarwgt = onc.variables[varname1]
-            ivarwgtv = iovarwgt[:]*onc.variables[varname2]
+            iovarwgt1 = onc.variables[varname1]
+            iovarwgt2 = onc.variables[varname2]
+            ivarwgtv = iovarwgt1[:]*iovarwgt2[:]
 
         if oper == 'inv':
@@ -17342 +17343 @@
                 for id1 in range(loopshape[0]):
                     slicevalues = SliceVar(iovar,dimsloop,[id1])
-                    slicewgt = SliceVar(iovarwgt,dimsloop,[id1])
-                    newvals[id1] = np.mean(ivarv[tuple(slicevalues)] /               \
-                      ivarwgtv[tuple(slicewgt)])
+                    slicewgt = SliceVar(iovarwgt1,dimsloop,[id1])
+                    TOTsumwgt = np.sum(1./ivarwgtv[tuple(slicewgt)])
+                    newvals[id1] = np.sum(ivarv[tuple(slicevalues)] /                \
+                      ivarwgtv[tuple(slicewgt)]) / TOTsumwgt
             elif len(loopshape) == 2:
                 newvals = np.ones((loopshape[0],loopshape[1]),dtype=np.float)*       \