Changeset 889 in lmdz_wrf for trunk

Timestamp:

Jun 18, 2016, 10:24:29 PM (8 years ago)

Author:

lfita

Message:

Working version of `operdim'

File:

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

trunk/tools/nc_var_tools.py

@@ -144 +144 @@
     def __init__(self,dx,dy,dz,dt,dnx,dny,dnz,dnt):
         totshape = dt*dz*dy*dx
-        dimensions = {}
         
         self.values = np.arange((totshape), dtype=np.float).reshape(dt,dz,dy,dx)
-        self.dimensions = {dnx: dx, dny: dy, dnz: dz, dnt: dt}
+        self.dimensionsdict = {dnt: dt, dnz: dz, dny: dy, dnx: dx}
+        self.dimensions = [dnt, dnz, dny, dnx]
         self.shape = (dt, dz, dy, dx)
         self.standard_name = 'general_var'
@@ -8785 +8785 @@
       doper: names of the dimensions to operate
       oper: operation to perform
+    >>> variable = genericNCvariable(5,4,3,2,'lon','lat','pres','time')
+    >>> varv = variable[:]
+    >>> vardims = variable.dimensions
+    >>> doper = ['time', 'pres']
+    >>> oper = 'mean'
+    >>> operv, operd = operdim(varv, vardims, doper, oper)
+    [[ 50.  51.  52.  53.  54.]
+     [ 55.  56.  57.  58.  59.]
+     [ 60.  61.  62.  63.  64.]
+     [ 65.  66.  67.  68.  69.]]
+    ['lat', 'lon']
     """ 
     fname = 'operdim'
@@ -8795 +8806 @@
             operslice.append(True)
 
-    operddims = []
+    opereddims = []
     noper = 0
+    varoper = varv.copy()
     for idim in range(len(varv.shape)):
+        print 'Lluis:', idim, vardims[idim], operslice[idim],':',varoper.shape
         if operslice[idim]:
             if oper == 'max':
-                varoper = np.max(varv, axis=idim-noper)
+                varoper = np.max(varoper, axis=idim-noper)
             elif oper == 'mean':
-                varoper = np.mean(varv, axis=idim-noper)
+                varoper = np.mean(varoper, axis=idim-noper)
             elif oper == 'mean2':
-                varoper = np.mean(varv, axis=idim-noper)
+                varoper = np.mean(varoper, axis=idim-noper)
             elif oper == 'min':
-                varoper = np.min(varv, axis=idim-noper)
+                varoper = np.min(varoper, axis=idim-noper)
             elif oper == 'sum':
-                varoper = np.sum(varv, axis=idim-noper)
+                varoper = np.sum(varoper, axis=idim-noper)
             else:
                 print errormsg
@@ -8815 +8828 @@
             noper = noper + 1
         else:
-            operddims.append(vardims[idim])
+            opereddims.append(vardims[idim])
+
+    operedvals = varoper
             
     return operedvals, opereddims
-
-variable = genericNCvariable(5,4,3,2,'lon','lat','pres','time')
-vv = variable[:]
-vardims = variable.dimensions
-doper = 'time'
-oper = 'mean'
-
-operv, operd = operdim(varv, vardims, doper, oper)
-print operv
-print operd
-quit()
 
 def operation_alongdims(ovar,dimvals,dimsoper,opkind):