Changeset 2298 in lmdz_wrf

Timestamp:

Jan 29, 2019, 7:54:10 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `remove_monotones': Function to remove the monotones (len(dim) = 1) from an array

File:

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

trunk/tools/generic_tools.py

@@ -171 +171 @@
 # radius_dist: Function to generate a matrix with the distance at a given point
 # rectangular_spiral: Function to provide a rectangular spiral (along x,y-axis values) of values
+# remove_monotones: Function to remove the monotones (len(dim) = 1) from an array
 # replace_list: Function to replace a value in a given list
 # roman_to_int: Convert a roman numeral to an integer
@@ -15048 +15049 @@
     return strlist
 
+def remove_monotones(array, dimns):
+    """ Function to remove the monotones (len(dim) = 1) from an array
+      array: array to remove the monotones
+      dimns: list with the names of the dimensions of the array
+    >>> remove_monotones(np.arange(20).reshape(4,1,5), ['dx', 'dy', 'dz'])
+    (array([[ 0,  1,  2,  3,  4],
+           [ 5,  6,  7,  8,  9],
+           [10, 11, 12, 13, 14],
+           [15, 16, 17, 18, 19]]), ['dx', 'dz'])
+    """
+    fname = 'remove_monotones'
+
+    rank = len(array.shape)
+
+    newshape = []
+    newdims = []
+    slicev = []
+    for idd in range(rank):
+        if array.shape[idd] == 1: slicev.append(0)
+        else: 
+            newshape.append(array.shape[idd])
+            slicev.append(slice(0,array.shape[idd]))
+            newdims.append(dimns[idd])
+
+    newarray = np.zeros(tuple(newshape), dtype=array.dtype)
+    newarray = array[tuple(slicev)]
+
+    return newarray, newdims
+
 #quit()