Only try to extend rii and cii when not empty

jenshnielsen · jenshnielsen · commit ba1e33906f80 · 2015-08-01T20:13:26.000+01:00
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -1750,19 +1750,19 @@ def plot_wireframe(self, X, Y, Z, *args, **kwargs):
 
         if rstride:
             rii = list(xrange(0, rows, rstride))
+            # Add the last index only if needed
+            if rows > 0 and rii[-1] != (rows - 1) :
+                rii += [rows-1]
         else:
             rii = []
-        if cstride:                
+        if cstride:
             cii = list(xrange(0, cols, cstride))
+            # Add the last index only if needed
+            if cols > 0 and cii[-1] != (cols - 1) :
+                cii += [cols-1]
         else:
             cii = []
 
-        # Add the last index only if needed
-        if rows > 0 and rii[-1] != (rows - 1) :
-            rii += [rows-1]
-        if cols > 0 and cii[-1] != (cols - 1) :
-            cii += [cols-1]
-
         # If the inputs were empty, then just
         # reset everything.
         if Z.size == 0 :
