Patched pyplot.subplots for squeeze==False

jseabold · jseabold · commit 551d35c17b31 · 2011-06-02T17:39:47.000-04:00
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -799,6 +799,9 @@ def subplots(nrows=1, ncols=1, sharex=False, sharey=False, squeeze=True,
             ret = fig, axarr[0,0]
         else:
             ret = fig, axarr.squeeze()
+    else:
+        # returned axis array will be always 2-d, even if nrows=ncols=1
+        return fig, axarr.reshape(nrows, ncols)
 
 
     return ret
