Changeset 2271 in lmdz_wrf for trunk/tools/nc_var_tools.py

Timestamp:

Dec 27, 2018, 8:46:52 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `area_weighted': Function to provide an area weighted statistics within two different values of matrices assuming regular lon/lat projections without taking into account Earth's curvature

File:

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

trunk/tools/nc_var_tools.py

@@ -28 +28 @@
 # addVar: Operation to add a new variable in a file using exisintg dimensions
 # addvals: Function to add values to a given variable at a given dimension
+# area_weighted: Function to provide an area weighted statistics within two different values 
+#   of matrices assuming regular lon/lat projections without taking into account Earth's curvature
 # attrinvar: Check existence of an attribute in a netCDF variable object
 # basicvardef: Function to give the basic attributes to a variable (std_name, long_name, units)
@@ -27222 +27224 @@
         quit()
 
-    statsavail = ['mean', 'min', 'max', 'mean2', 'stddev']
+    statsavail = ['mean', 'min', 'max', 'mean2', 'stddev', 'count']
     # Characteristics of the statistics
     statnschars = {'mean': ['spcmean', 'spatial mean', 1],                           \
@@ -27238 +27240 @@
 
     for stn in statns:
-        if not gen.searchInlist(statsavail, stn):
+        if not gen.searchInlist(statnschars.keys(), stn):
             print errormsg
             print '  ' + fname + ": statistics '" + stn + "' not ready !!"
-            print '    available ones:', statnschar.keys()
+            print '    available ones:', statsavail
             quit(-1)
 
@@ -27574 +27576 @@
             varns.append(vn)
 
+    varrm = [refvarxn, refvaryn, refvarxbndsn, refvarybndsn, getvarxn, getvaryn,     \
+      getvarxbndsn, getvarybndsn]
+    for vn in varrm: 
+        if gen.searchInlist(varns, vn): varns.remove(vn)
+
     for vn in varns:
         print '  ' + fname + ": using '" + vn + "' ..."
@@ -27587 +27594 @@
 
             if stn == 'count': 
-                svals = np.unique(varin)
+                svals = np.unique(varvals)
                 statchar[2] = len(svals)
 
-            varout = fsci.module_scientific.spaceweightstats(varin=varvalst,         \
+            varoutt = fsci.module_scientific.spaceweightstats(varin=varvalst,        \
               ngridsin=Ngridsint, gridsin=gridsint, percentages=percenst,            \
               stats=stn, dxa=refdx, dya=refdy, dxb=getdx, dyb=getdy,                 \
               maxngridsin=Ngridsinmax, lstats=statchar[2])
 
-            varout = varout.transpose()
+            varout = varoutt.transpose()
             if statchar[2] == 1:
                 newvar = onewnc.createVariable(vn + statchar[0], 'f', ('lat','lon'))
@@ -27601 +27608 @@
             else:
                 if stn == 'count':
-                    if not gen.searchInlist(onewnc.dimensions, 'count'): 
-                        newdim = onewnc.createDimension('count', statchar[2])
-                        newvar = onewnc.createVariable(vn + 'count', 'f', ('count'))
+                    if not gen.searchInlist(onewnc.dimensions, vn+'count'): 
+                        newdim = onewnc.createDimension(vn+'count', statchar[2])
+                        newvar = onewnc.createVariable(vn+'count', 'f', (vn+'count'))
                         newvar[:] = svals
                         basicvardef(newvar, vn+'count', 'unique .get. values of ' +  \
                           vn, ovar.units)
 
-                    newvar = onewnc.createVariable(vn + statchar[0], 'f', ('count',  \
+                    newvar = onewnc.createVariable(vn+statchar[0], 'f', (vn+'count', \
                       'lat','lon'))
                     newvar[:] = varout[:]
@@ -27627 +27634 @@
     return
 
-fvals= 'reference_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1,get_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1'
-area_weighted('yes:min,max,mean,stddev',fvals,'ct_values')
+#fvals= 'reference_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1,get_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1'
+#area_weighted('yes:min,max,mean,stddev',fvals,'ct_values')