Changeset 2218 in lmdz_wrf

Timestamp:

Nov 7, 2018, 4:43:47 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `all_consecutive_combs': Function to provide all the consecutive possible combinations from a series of indices

File:

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

trunk/tools/generic_tools.py

@@ -59 +59 @@
 # advance_date: Function to advance a date object a certain given units of time
 # advance_matDate: Function to advance matrix-date with a matrix-increment
+# all_consecutive_combs: Function to provide all the consecutive possible 
+#   combinations from a series of indices
 # angle_DegMinSec: Function to transform an angle to Degrees Minutes Seconds
 # ASCIIfile_stats: Function to provide the statistics of a series of values from an ASCII file
@@ -14727 +14729 @@
     return colors
 
+def all_consecutive_combs(sizes):
+    """ Function to provide all the consecutive possible combinations from a series 
+      of indices
+      sizes: list of indices to expand and combine
+    >>> all_consecutive_combs([3, 2, 2])
+    [[0 0 0]
+     [0 0 1]
+     [0 1 0]
+     [0 1 1]
+     [1 0 0]
+     [1 0 1]
+     [1 1 0]
+     [1 1 1]
+     [2 0 0]
+     [2 0 1]
+     [2 1 0]
+     [2 1 1]]
+    """
+    fname = 'all_consecutive_combs'
+
+    Nsizes = len(sizes)
+    Ncombs = np.prod(sizes)
+    combs = np.zeros((Ncombs, Nsizes), dtype=int)
+
+    itervals = np.zeros((Nsizes), dtype=int)
+    for icomb in range(Ncombs):
+        combs[icomb,:] = itervals
+        overlimit = True
+        for i in range(Nsizes):
+            ind = Nsizes-i-1
+            itervals[ind] = itervals[ind] + 1
+            if itervals[ind] < sizes[ind]: overlimit = False
+            if not overlimit:
+                break
+            else:
+                itervals[ind] = 0
+
+    return combs
+
+#print all_consecutive_combs([3, 2, 2])
+
 #quit()