FIX: allow reshape 2-D to return a bare 1-d list

jklymak · jklymak · commit ec18892d56f6 · 2018-08-23T11:20:41.000-07:00
diff --git a/lib/matplotlib/cbook/__init__.py b/lib/matplotlib/cbook/__init__.py
@@ -1397,8 +1397,11 @@ def _reshape_2D(X, name):
         # 1D array of scalars: directly return it.
         return [X]
     elif X.ndim in [1, 2]:
-        # 2D array, or 1D array of iterables: flatten them first.
-        return [np.reshape(x, -1) for x in X]
+        if hasattr(X[0], '__len__'):
+            # 2D array, or 1D array of iterables: flatten them first.
+            return [np.reshape(x, -1) for x in X]
+        else:
+            return [X]
     else:
         raise ValueError("{} must have 2 or fewer dimensions".format(name))
 
