Replaced a loop with a map

mp4096 · mp4096 · commit 4d70d0025586 · 2016-11-20T21:08:11.000+01:00
* The performance remains the same.
* The source code is more readable.
diff --git a/control/statesp.py b/control/statesp.py
@@ -124,12 +124,7 @@ def __init__(self, *args):
 
         # Here we're going to convert inputs to matrices, if the user gave a
         # non-matrix type.
-        #! TODO: [A, B, C, D] = map(matrix, [A, B, C, D])?
-        matrices = [A, B, C, D]
-        for i in range(len(matrices)):
-            # Convert to matrix first, if necessary.
-            matrices[i] = matrix(matrices[i])
-        [A, B, C, D] = matrices
+        A, B, C, D = map(matrix, [A, B, C, D])
 
         LTI.__init__(self, B.shape[1], C.shape[0], dt)
         self.A = A
