fix to init() to handle 0d arrays properly

murrayrm · murrayrm · commit 020734dc91f5 · 2016-12-26T06:42:26.000-08:00
diff --git a/control/xferfcn.py b/control/xferfcn.py
@@ -125,7 +125,9 @@ def __init__(self, *args):
         # but be careful.
         data = [num, den]
         for i in range(len(data)):
-            if isinstance(data[i], (int, float, complex)):
+            # Check for a scalar (including 0d ndarray)
+            if (isinstance(data[i], (int, float, complex)) or
+                (isinstance(data[i], ndarray) and data[i].ndim == 0)):
                 # Convert scalar to list of list of array.
                 if (isinstance(data[i], int)):
                     # Convert integers to floats at this point
