TR updates, first round

Author: lpozo

python-flatten-list/flatten.py

@@ -3,8 +3,11 @@
 from operator import add, concat, iconcat
 
 
-def flatten_comprehension(matrix):
-    return [item for row in matrix for item in row]
+def flatten_extend(matrix):
+    flat = []
+    for row in matrix:
+        flat.extend(row)
+    return flat
 
 
 def flatten_concatenation(matrix):
@@ -14,17 +17,18 @@ def flatten_concatenation(matrix):
     return flat
 
 
-def flatten_extend(matrix):
-    flat = []
-    for row in matrix:
-        flat.extend(row)
-    return flat
+def flatten_comprehension(matrix):
+    return [item for row in matrix for item in row]
 
 
-def flatten_itertools_chain(matrix):
+def flatten_chain(matrix):
     return list(chain.from_iterable(matrix))
 
 
+def flatten_reduce_lambda(matrix):
+    return list(reduce(lambda x, y: x + y, matrix, []))
+
+
 def flatten_reduce_add(matrix):
     return reduce(add, matrix, [])
 
@@ -37,9 +41,5 @@ def flatten_reduce_iconcat(matrix):
     return reduce(iconcat, matrix, [])
 
 
-def flatten_reduce_lambda(matrix):
-    return list(reduce(lambda x, y: x + y, matrix, []))
-
-
 def flatten_sum(matrix):
     return sum(matrix, [])

python-flatten-list/performance.py

@@ -6,15 +6,15 @@
 TO_MS = 1000
 NUM = 10
 FUNCTIONS = [
-    "flatten_comprehension",
-    "flatten_concatenation",
     "flatten_extend",
+    "flatten_concatenation",
+    "flatten_comprehension",
+    "flatten_chain",
     "flatten_reduce_lambda",
-    "flatten_sum",
-    "flatten_itertools_chain",
     "flatten_reduce_add",
     "flatten_reduce_concat",
     "flatten_reduce_iconcat",
+    "flatten_sum",
 ]
 
 matrix = [list(range(SIZE))] * SIZE
@@ -27,4 +27,4 @@
 print(f"Time to flatten a {SIZE}x{SIZE} matrix (in milliseconds):\n")
 
 for func, time in sorted(results.items(), key=lambda result: result[1]):
-    print(f"{func + '()':.<30}{time * TO_MS / NUM:.>10.6f} ms")
+    print(f"{func + '()':.<30}{time * TO_MS / NUM:.>7.2f} ms")