Fix list creation

jklymak · timhoffm · web-flow · commit c151e71d39d7 · 2022-01-04T09:11:45.000+01:00
Co-authored-by: Tim Hoffmann &lt;2836374+timhoffm@users.noreply.github.com&gt;
diff --git a/lib/matplotlib/colorbar.py b/lib/matplotlib/colorbar.py
@@ -1414,8 +1414,8 @@ def make_axes(parents, location=None, orientation=None, fraction=0.15,
     # use .flatten or .ravel as these copy the references rather than
     # reuse them, leading to a memory leak
     if isinstance(parents, np.ndarray):
-        parents = [parent for parent in parents.flat]
-    if not isinstance(parents, list):
+        parents = list(parents.flat)
+    elif not isinstance(parents, list):
         parents = [parents]
     fig = parents[0].get_figure()
 
