Use output.shape

mdboom · mdboom · commit d21a6edc9fa5 · 2016-02-17T09:47:57.000-05:00
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -359,8 +359,10 @@ def _make_image(self, A, in_bbox, out_bbox, clip_bbox, magnification=1.0,
             subset = TransformedBbox(
                 clip_bbox, t0.frozen().inverted()).frozen()
             output = output[
-                int(max(subset.ymin, 0)):int(min(subset.ymax + 1, A.shape[0])),
-                int(max(subset.xmin, 0)):int(min(subset.xmax + 1, A.shape[1]))]
+                int(max(subset.ymin, 0)):
+                int(min(subset.ymax + 1, output.shape[0])),
+                int(max(subset.xmin, 0)):
+                int(min(subset.xmax + 1, output.shape[1]))]
 
             t = Affine2D().translate(
                 int(max(subset.xmin, 0)), int(max(subset.ymin, 0))) + t
