Merge pull request #133 from jseabold/fix-subplot-squeeze

ddale · ddale · commit ac02f8382362 · 2011-06-16T08:03:14.000-07:00
Patched pyplot.subplots for squeeze==False
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -829,7 +829,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
+        ret = fig, axarr.reshape(nrows, ncols)
 
     return ret
 
