MNT: simplify creation of output array

tacaswell · tacaswell · commit 80909e5b7e5f · 2017-03-14T18:19:45.000-04:00
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -390,8 +390,6 @@ def _make_image(self, A, in_bbox, out_bbox, clip_bbox, magnification=1.0,
                         mask = A.mask
                     # ~A.mask  # masked data
                     A = rgba
-                    output = np.zeros((out_height, out_width, 4),
-                                      dtype=A.dtype)
                     alpha = 1.0
                     created_rgba_mask = True
                 else:
@@ -414,12 +412,12 @@ def _make_image(self, A, in_bbox, out_bbox, clip_bbox, magnification=1.0,
                 elif A.shape[2] != 4:
                     raise ValueError("Invalid dimensions, got %s" % (A.shape,))
 
-                output = np.zeros((out_height, out_width, 4), dtype=A.dtype)
-
                 alpha = self.get_alpha()
                 if alpha is None:
                     alpha = 1.0
 
+            output = np.zeros((out_height, out_width, 4), dtype=A.dtype)
+
             _image.resample(
                 A, output, t, _interpd_[self.get_interpolation()],
                 self.get_resample(), alpha,
