Changeset 1699 in lmdz_wrf

Timestamp:

Dec 8, 2017, 5:42:47 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `fill_Narray': Function to fill a n-dimensional array with an arrary of lesser rank

File:

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

trunk/tools/generic_tools.py

@@ -79 +79 @@
 # dtsec360dyr: Class to operate a number of seconds to a date in a 360 days/yr (or 12 30-days months) calendar
 # files_folder_HMT: Function to retrieve a list of files from a folder [fold] and files with [head]*[middle]*[tail]
+# fill_Narray: Function to fill a n-dimensional array with an arrary of lesser rank
 # get_configuration: Function to get the configuration from an ASCII external file
 # get_specdictionary_HMT: Function to get specific values from a dictionary by selcting that keys with H*M*T
@@ -12485 +12486 @@
     return True
 
+def fill_Narray(idata, narray, filldim=None):
+    """ Function to fill a n-dimensional array with an arrary of lesser rank
+      idata: initial array
+      narray: enpty narray yo fill
+      filldim: which number of dimensions have to be filled (None, for automatic)
+        provide a list of number of dimensions in case some lengths of dimensions are repeated
+      >>> ivec = np.arange(3, dtype=int)
+      >>> nmat = np.zeros((3,2), dtype=int)
+      >>> fill_Narray(ivec, nmat)
+      [[0 0]
+       [1 1]
+       [2 2]]
+      >>> nmat = np.zeros((3,2,3), dtype=int)
+      >>> fill_Narray(ivec, nmat, filldim=[2])
+      [[[0 0 0]
+        [0 0 0]]
+
+       [[1 1 1]
+        [1 1 1]]
+
+       [[2 2 2]
+        [2 2 2]]]
+    """
+    fname = 'fill_Narray'
+
+    ishape = idata.shape
+    nshape = narray.shape
+
+    Ndimsi = len(ishape)
+    Ndimsn = len(nshape)
+
+    # Dictionary with the equivalencies among dimensions 
+    #    (-1, fill; 0, same dim as input)
+    dimequiv = {}
+    dimns = []
+    rundims = []
+    for idim in range(Ndimsn):
+        dimns.append(str(idim))
+        if searchInlist(ishape, nshape[idim]):
+            if filldim is not None:
+                if searchInlist(filldim, idim): 
+                    dimequiv[idim] = -1
+                    rundims.append(str(idim)) 
+                else: dimequiv[idim] = 0
+            else:
+                dimequiv[idim] = 0
+        else:
+            dimequiv[idim] = -1
+            rundims.append(str(idim)) 
+
+    # Getting all the slices to fill
+    narrayslices = provide_slices(dimns, list(nshape), rundims)
+
+    # Filling narray
+    Nslices = len(narrayslices)
+    for islc in range(Nslices):
+        narray[tuple(narrayslices[islc])] = idata
+
+    return narray
+
 #quit()