Changeset 2216 in lmdz_wrf

Timestamp:

Nov 6, 2018, 8:22:44 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `multi_index_vec': Function to provide the coordinates of multiples repetitions of a value inside a vector

File:

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

trunk/tools/generic_tools.py

@@ -146 +146 @@
 # multi_index_mat: Function to provide the multiple coordinates of a given value inside a matrix
 # multi_index_string: Function to provide the indeces of every repetition of a group of characters within a string
+# multi_index_vec: Function to provide the coordinates of multiples repetitions of a value inside a vector
 # Nomasked: Function to bring back a given array wthout the masked values reducing a given dimension
 # Nstr: Function to transform a number to string, but making sure it preserves 
@@ -2385 +2386 @@
     return valpos
 
+def multi_index_vec(vec,val):
+    """ Function to provide the coordinates of multiples repetitions of a value 
+      inside a vector
+    index_vec(vec,val)
+      vec= vector with values
+      val= value to search
+    >>> vecv = np.arange(27)
+    >>> vecv[2] = 22
+    >>> vecv[12] = 22
+    >>> multi_index_vec(vecv,22)
+    2, 12, 22
+    """
+    fname = 'multi_index_vec'
+
+    valpos = []
+    if type(vec) == type(np.arange(2)):
+        for i in range(vec.shape[0]):
+            if vec[i] == val: valpos.append(i)                
+    elif type(vec) == type(tuple([1,2])):
+        for i in range(len(vec)):
+            if vec[i] == val: valpos.append(i)
+    elif type(vec) == type(range(2)):
+        for i in vec:
+            if vec[i] == val: valpos.append(i)
+    else:
+        print errormsg
+        print '  ' + fname + ': type', type(vec), 'not ready !!'
+        print '    available ones:', type(np.arange(2)), type(range(2)),             \
+          type(tuple([1,2]))
+        quit(-1)
+
+    if len(valpos) == 0: valpos.append(-1)
+
+    return valpos
+
 def index_mat(mat,val,quitval=True):
     """ Function to provide the coordinates of a given value inside a matrix