More code review updates

scottshambaugh · scottshambaugh · commit 74543da8cbf2 · 2025-01-08T09:03:04.000-07:00
diff --git a/lib/matplotlib/transforms.py b/lib/matplotlib/transforms.py
@@ -945,6 +945,8 @@ def update_from_data_x(self, x, ignore=None):
            - When ``None``, use the last value passed to :meth:`ignore`.
         """
         x = np.ravel(x)
+        # The y-component in np.array([x, *y*]).T is not used. We simply pass
+        # x again to not spend extra time on creating an array of unused data
         self.update_from_data_xy(np.array([x, x]).T, ignore=ignore, updatey=False)
 
     def update_from_data_y(self, y, ignore=None):
@@ -963,6 +965,8 @@ def update_from_data_y(self, y, ignore=None):
             - When ``None``, use the last value passed to :meth:`ignore`.
         """
         y = np.ravel(y)
+        # The x-component in np.array([*x*, y]).T is not used. We simply pass
+        # y again to not spend extra time on creating an array of unused data
         self.update_from_data_xy(np.array([y, y]).T, ignore=ignore, updatex=False)
 
     def update_from_data_xy(self, xy, ignore=None, updatex=True, updatey=True):
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -2413,29 +2413,34 @@ def plot_wireframe(self, X, Y, Z, *, axlim_clip=False, **kwargs):
                 rstride = int(max(np.ceil(rows / rcount), 1)) if rcount else 0
                 cstride = int(max(np.ceil(cols / ccount), 1)) if ccount else 0
 
+        if rstride == 0 and cstride == 0:
+            raise ValueError("Either rstride or cstride must be non zero")
+
         # We want two sets of lines, one running along the "rows" of
         # Z and another set of lines running along the "columns" of Z.
         # This transpose will make it easy to obtain the columns.
         tX, tY, tZ = np.transpose(X), np.transpose(Y), np.transpose(Z)
 
-        if rstride:
+        # Compute the indices of the row and column lines to be drawn
+        if rstride == 0:
+            rii = np.array([], dtype=int)
+        elif (rows - 1) % rstride == 0:
+            # last index is hit: rii[-1] == rows - 1
             rii = np.arange(0, rows, rstride)
-            # Add the last index only if needed
-            if rows > 0 and rii[-1] != (rows - 1):
-                rii = np.append(rii, rows-1)
         else:
-            rii = np.array([], dtype=int)
+            # add the last index
+            rii = np.arange(0, rows + rstride, rstride)
+            rii[-1] = rows - 1
 
-        if cstride:
+        if cstride == 0:
+            cii = np.array([], dtype=int)
+        elif (cols - 1) % cstride == 0:
+            # last index is hit: cii[-1] == cols - 1
             cii = np.arange(0, cols, cstride)
-            # Add the last index only if needed
-            if cols > 0 and cii[-1] != (cols - 1):
-                cii = np.append(cii, cols-1)
         else:
-            cii = np.array([], dtype=int)
-
-        if rstride == 0 and cstride == 0:
-            raise ValueError("Either rstride or cstride must be non zero")
+            # add the last index
+            cii = np.arange(0, cols + cstride, cstride)
+            cii[-1] = cols - 1
 
         # If the inputs were empty, then just
         # reset everything.
