Changeset 1963 in lmdz_wrf

Timestamp:

Jul 24, 2018, 5:42:09 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `stats_lonlat_box': Function to provide the statistics of a variable from a lon,lat box

File:

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

trunk/tools/nc_var_tools.py

@@ -151 +151 @@
 # splitfile_dim: Function to split a file along a given dimension with a new file for each different value along the dimension (assuming resultant vardim of rank-1)
 # statcompare_files: Python script to statistically compare two different files
+# stats_lonlat_box: Function to provide the statistics of a variable from a lon,lat box
 # submns: Function to retrieve a series of months from a file
 # subyrs: Function to retrieve a series of years from a file
@@ -1462 +1463 @@
       print errormsg
       print '   ' + fname + ': File "' + refnc + '" does not have variable: ' + refvar
+      refvarssort = ncref.variables.keys()
+      refvarssort.sort()
+      print '    available ones:', refvarssort
       ncf.close()
       ncref.close()
@@ -24824 +24828 @@
     return
 
+def stats_lonlat_box(vals, lonv, latv, nlon, xlon, nlat, xlat):
+    """ Function to provide the statistics of a variable from a lon,lat box
+      vals= values (2D matrix)
+      lonv= longitude  values(2D matrix)
+      latv= latitude values (2D matrix)
+      nlon,xlon= minimum and maximum of longitudes
+      nlat,xlat= minimum and maximum of latitudes
+        * Returns: [minval, maxval, meanval, stdval], [isw, jsw, ine, jne]
+      >>> mat = np.arange(40).reshape(10,4)
+      >>> lon1D = np.arange(-5.,5.,1)
+      >>> lat1D = np.arange(-35.,-31.,1.)
+      >>> lat2D, lon2D = np.meshgrid(lat1D, lon1D)
+      >>> stats_lonlat_box(mat, lon2D, lat2D, -4., 0., -35., -33.)
+      [4, 39, 21.5, 10.388294694831615] [0, 1, 5, 11]
+    """
+    fname = 'stats_lonlat_box'
+
+    # Distance to vertex
+    SWdist = np.sqrt((lonv-nlon)**2 + (latv-nlat)**2)
+    NEdist = np.sqrt((lonv-xlon)**2 + (latv-xlat)**2)
+
+    nSWdist = np.min(SWdist)
+    nNEdist = np.min(NEdist)
+
+    # indices of vertices
+    jsw, isw = gen.index_mat(SWdist, nSWdist)
+    jne, ine = gen.index_mat(NEdist, nNEdist)
+
+    ine = np.max([ine+1, vals.shape[1]+1])
+    jne = np.max([jne+1, vals.shape[0]+1])
+
+    minval = np.min(vals[jsw:jne,isw:ine])
+    maxval = np.max(vals[jsw:jne,isw:ine])
+    meanval = np.mean(vals[jsw:jne,isw:ine])
+    stdval = np.std(vals[jsw:jne,isw:ine])
+
+    return [minval, maxval, meanval, stdval], [isw, jsw, ine, jne]
+
 #quit()