Changeset 709 in lmdz_wrf for trunk

Timestamp:

Apr 20, 2016, 6:00:26 PM (9 years ago)

Author:

lfita

Message:

Adding `multi_index_mat': Function to provide the multiple coordinates of a given value inside a matrix

File:

• trunk/tools/nc_var_tools.py (modified) (1 diff)

trunk/tools/nc_var_tools.py

@@ -1210 +1210 @@
     return valpos
 
+def multi_index_mat(mat,val):
+    """ Function to provide the multiple coordinates of a given value inside a matrix
+    index_mat(mat,val)
+      mat= matrix with values
+      val= value to search
+    >>> vals = np.ones((24), dtype=np.float).reshape(2,3,4)
+    vals[:,:,2] = 0.
+    vals[1,:,:] = np.pi
+    vals[:,2,:] = -1.
+    multi_index_mat(vals,1.)
+    [array([0, 0, 0]), array([0, 0, 1]), array([0, 0, 3]), array([0, 1, 0]), array([0, 1, 1]), array([0, 1, 3])]
+    """
+    fname = 'multi_index_mat'
+
+    matshape = mat.shape
+
+    ivalpos = []
+    matlist = list(mat.flatten())
+    Lmatlist = len(matlist)
+    
+    val0 = val - val
+    if val != val0:
+        valdiff = val0
+    else:
+        valdiff = np.ones((1), dtype = type(val))
+    
+    ifound = 0
+    while ifound < Lmatlist:
+        if matlist.count(val) == 0:
+            ifound = Lmatlist + 1
+        else:
+            ifound = matlist.index(val)
+
+            Ndims = len(matshape)
+            valpos = np.zeros((Ndims), dtype=int)
+            baseprevdims = np.zeros((Ndims), dtype=int)
+
+            for dimid in range(Ndims):
+                baseprevdims[dimid] = np.product(matshape[dimid+1:Ndims])
+                if dimid == 0:
+                    alreadyplaced = 0
+                else:
+                    alreadyplaced = np.sum(baseprevdims[0:dimid]*valpos[0:dimid])
+                valpos[dimid] = int((ifound - alreadyplaced )/ baseprevdims[dimid])
+            matlist[ifound] = valdiff
+            ivalpos.append(valpos)
+
+    return ivalpos
+
 def addfileInfile(origfile,destfile,addfile,addsign):
     """ Function to add the content of a file [addfile] to another one [origfile] at