Changeset 13193 for main/waeup.uniben/trunk/src/waeup

Timestamp:

9 Aug 2015, 22:46:04 (9 years ago)

Author:

uli

Message:

Fix partition().

Return last element with proper end value.

File:

• main/waeup.uniben/trunk/src/waeup/uniben/scripts.py (modified) (1 diff)

main/waeup.uniben/trunk/src/waeup/uniben/scripts.py

@@ -142 +142 @@
 
 def partition(container, part_size=10000):
-    """Return a list of triples (<num>, <start>, <stop>).
+    """Partition `container` into chunks.
+
+    Get a list of triples (<num>, <index_start>, <index_end>) which
+    represent chunks of elements from `container`.
+
+    The `container` object must support `len()`.
+
+    Split length of `container` and tell what partitions we get, if each
+    partition is size `part_size` or less.
+
+    For instance a container of size 250 and `part_size` 100 would give:
+
+      [(0,   0,  99),
+       (1, 100, 199),
+       (2, 200, 249),
+       ]
 
     """
     num = len(container)
-    print("Number of studs: %s" % num)
-    return [(idx, start, start + part_size - 1) for idx, start in enumerate(
-        range(0, num, part_size))]
+    print("Container elements: %s" % num)
+    return [(idx, start, min(start + part_size - 1, num - 1))
+             for idx, start in enumerate(range(0, num, part_size))]