Small cleanup

apaszke · apaszke · commit 168ce3acedcd · 2020-03-06T14:46:09.000+01:00
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -1566,12 +1566,12 @@ def plot_surface(self, X, Y, Z, *args, norm=None, vmin=None,
         cpadding = cstride - crem if crem != 0 else 0
 
         def pad_for_patches(a):
+            if cpadding == 0 and rpadding == 0:
+                return a
             result = np.empty_like(a, shape=(rows + rpadding, cols + cpadding))
             result[:rows, :cols] = a
-            if cpadding:
-                result[:rows, -cpadding:] = a[:, -1:]
-            if rpadding:
-                result[-rpadding:, :cols] = a[-1:, :]
+            result[:rows, cols:] = a[:, -1:]
+            result[rows:, :cols] = a[-1:, :]
             result[rows:, cols:] = a[-1, -1]
             return result
 
