allow _ssmatrix to accept complex values

lytex · lytex · commit e6f2f60f6061 · 2020-02-09T21:52:35.000+01:00
diff --git a/control/statesp.py b/control/statesp.py
@@ -93,9 +93,15 @@ def _ssmatrix(data, axis=1):
     # Convert the data into an array or matrix, as configured
     # If data is passed as a string, use (deprecated?) matrix constructor
     if config.defaults['statesp.use_numpy_matrix'] or isinstance(data, str):
-        arr = np.matrix(data, dtype=float)
+        if np.isrealobj(data):
+            arr = np.matrix(data, dtype=float)
+        elif np.iscomplexobj(data):
+            arr = np.matrix(data, dtype=complex)
     else:
-        arr = np.array(data, dtype=float)
+        if np.isrealobj(data):
+            arr = np.array(data, dtype=float)
+        elif np.iscomplexobj(data):
+            arr = np.array(data, dtype=complex)
     ndim = arr.ndim
     shape = arr.shape
 
