Convert six.moves.xrange() to range() for Python 3

cclauss · cclauss · commit f73b62c6a16f · 2018-02-18T23:26:06.000+01:00
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -21,7 +21,7 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange, zip
+from six.moves import zip
 
 import abc
 import contextlib
@@ -1682,7 +1682,7 @@ def __init__(self, fig, func, frames=None, init_func=None, fargs=None,
             if hasattr(frames, '__len__'):
                 self.save_count = len(frames)
         else:
-            self._iter_gen = lambda: iter(xrange(frames))
+            self._iter_gen = lambda: iter(range(frames))
             self.save_count = frames
 
         if self.save_count is None:
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -4,7 +4,6 @@
 from collections import OrderedDict
 
 import six
-from six.moves import xrange
 
 import itertools
 import warnings
@@ -393,7 +392,7 @@ def _plot_args(self, tup, kwargs):
         if ncx > 1 and ncy > 1 and ncx != ncy:
             cbook.warn_deprecated("2.2", "cycling among columns of inputs "
                                   "with non-matching shapes is deprecated.")
-        for j in xrange(max(ncx, ncy)):
+        for j in range(max(ncx, ncy)):
             seg = func(x[:, j % ncx], y[:, j % ncy], kw, kwargs)
             ret.append(seg)
         return ret
diff --git a/lib/matplotlib/backend_bases.py b/lib/matplotlib/backend_bases.py
@@ -36,7 +36,6 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange
 
 from contextlib import contextmanager
 from functools import partial
@@ -440,7 +439,7 @@ def _iter_collection_raw_paths(self, master_transform, paths,
             return
 
         transform = transforms.IdentityTransform()
-        for i in xrange(N):
+        for i in range(N):
             path = paths[i % Npaths]
             if Ntransforms:
                 transform = Affine2D(all_transforms[i % Ntransforms])
@@ -518,7 +517,7 @@ def _iter_collection(self, gc, master_transform, all_transforms,
             gc0.set_linewidth(0.0)
 
         xo, yo = 0, 0
-        for i in xrange(N):
+        for i in range(N):
             path_id = path_ids[i % Npaths]
             if Noffsets:
                 xo, yo = toffsets[i % Noffsets]
diff --git a/lib/matplotlib/backends/backend_svg.py b/lib/matplotlib/backends/backend_svg.py
@@ -5,7 +5,6 @@
 
 import six
 from six import unichr
-from six.moves import xrange
 
 import base64
 import codecs
@@ -1112,7 +1111,7 @@ def _draw_text_as_text(self, gc, x, y, s, prop, angle, ismath, mtext=None):
                 same_y = True
                 if len(chars) > 1:
                     last_y = chars[0][1]
-                    for i in xrange(1, len(chars)):
+                    for i in range(1, len(chars)):
                         if chars[i][1] != last_y:
                             same_y = False
                             break
diff --git a/lib/matplotlib/backends/backend_wx.py b/lib/matplotlib/backends/backend_wx.py
@@ -17,8 +17,6 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange
-import six
 
 import sys
 import os
@@ -1449,7 +1447,7 @@ def updateAxes(self, maxAxis):
             for menuId in self._axisId[maxAxis:]:
                 self._menu.Delete(menuId)
             self._axisId = self._axisId[:maxAxis]
-        self._toolbar.set_active(list(xrange(maxAxis)))
+        self._toolbar.set_active(list(range(maxAxis)))
 
     def getActiveAxes(self):
         """Return a list of the selected axes."""
diff --git a/lib/matplotlib/cbook/__init__.py b/lib/matplotlib/cbook/__init__.py
@@ -9,7 +9,7 @@
 from __future__ import absolute_import, division, print_function
 
 import six
-from six.moves import xrange, zip
+from six.moves import zip
 import bz2
 import collections
 import contextlib
@@ -941,7 +941,7 @@ def get_split_ind(seq, N):
 
     s_len = 0
     # todo: use Alex's xrange pattern from the cbook for efficiency
-    for (word, ind) in zip(seq, xrange(len(seq))):
+    for (word, ind) in zip(seq, range(len(seq))):
         s_len += len(word) + 1  # +1 to account for the len(' ')
         if s_len >= N:
             return ind
@@ -1098,7 +1098,7 @@ def allequal(seq):
     if len(seq) < 2:
         return True
     val = seq[0]
-    for i in xrange(1, len(seq)):
+    for i in range(1, len(seq)):
         thisval = seq[i]
         if thisval != val:
             return False
diff --git a/lib/matplotlib/colorbar.py b/lib/matplotlib/colorbar.py
@@ -22,7 +22,7 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange, zip
+from six.moves import zip
 
 import warnings
 
@@ -498,9 +498,9 @@ def _edges(self, X, Y):
         # Using the non-array form of these line segments is much
         # simpler than making them into arrays.
         if self.orientation == 'vertical':
-            return [list(zip(X[i], Y[i])) for i in xrange(1, N - 1)]
+            return [list(zip(X[i], Y[i])) for i in range(1, N - 1)]
         else:
-            return [list(zip(Y[i], X[i])) for i in xrange(1, N - 1)]
+            return [list(zip(Y[i], X[i])) for i in range(1, N - 1)]
 
     def _add_solids(self, X, Y, C):
         '''
@@ -561,9 +561,9 @@ def add_lines(self, levels, colors, linewidths, erase=True):
         x = np.array([0.0, 1.0])
         X, Y = np.meshgrid(x, y)
         if self.orientation == 'vertical':
-            xy = [list(zip(X[i], Y[i])) for i in xrange(N)]
+            xy = [list(zip(X[i], Y[i])) for i in range(N)]
         else:
-            xy = [list(zip(Y[i], X[i])) for i in xrange(N)]
+            xy = [list(zip(Y[i], X[i])) for i in range(N)]
         col = collections.LineCollection(xy, linewidths=linewidths)
 
         if erase and self.lines:
@@ -1337,7 +1337,7 @@ def _add_solids(self, X, Y, C):
         hatches = self.mappable.hatches * n_segments
 
         patches = []
-        for i in xrange(len(X) - 1):
+        for i in range(len(X) - 1):
             val = C[i][0]
             hatch = hatches[i]
 
diff --git a/lib/matplotlib/contour.py b/lib/matplotlib/contour.py
@@ -5,7 +5,6 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange
 
 import warnings
 import matplotlib as mpl
@@ -164,7 +163,7 @@ def clabel(self, *args, **kwargs):
         self.rightside_up = kwargs.get('rightside_up', True)
         if len(args) == 0:
             levels = self.levels
-            indices = list(xrange(len(self.cvalues)))
+            indices = list(range(len(self.cvalues)))
         elif len(args) == 1:
             levlabs = list(args[0])
             indices, levels = [], []
@@ -190,7 +189,7 @@ def clabel(self, *args, **kwargs):
             self.labelCValueList = np.take(self.cvalues, self.labelIndiceList)
         else:
             cmap = colors.ListedColormap(_colors, N=len(self.labelLevelList))
-            self.labelCValueList = list(xrange(len(self.labelLevelList)))
+            self.labelCValueList = list(range(len(self.labelLevelList)))
             self.labelMappable = cm.ScalarMappable(cmap=cmap,
                                                    norm=colors.NoNorm())
 
@@ -1340,7 +1339,7 @@ def find_nearest_contour(self, x, y, indices=None, pixel=True):
         # Nonetheless, improvements could probably be made.
 
         if indices is None:
-            indices = list(xrange(len(self.levels)))
+            indices = list(range(len(self.levels)))
 
         dmin = np.inf
         conmin = None
diff --git a/lib/matplotlib/hatch.py b/lib/matplotlib/hatch.py
@@ -6,7 +6,6 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange
 
 import numpy as np
 from matplotlib.path import Path
@@ -115,7 +114,7 @@ def set_vertices_and_codes(self, vertices, codes):
         shape_size = len(shape_vertices)
 
         cursor = 0
-        for row in xrange(self.num_rows + 1):
+        for row in range(self.num_rows + 1):
             if row % 2 == 0:
                 cols = np.linspace(0.0, 1.0, self.num_rows + 1, True)
             else:
diff --git a/lib/matplotlib/markers.py b/lib/matplotlib/markers.py
@@ -87,7 +87,6 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange
 
 from collections import Sized
 from numbers import Number
@@ -101,7 +100,7 @@
 # special-purpose marker identifiers:
 (TICKLEFT, TICKRIGHT, TICKUP, TICKDOWN,
  CARETLEFT, CARETRIGHT, CARETUP, CARETDOWN,
- CARETLEFTBASE, CARETRIGHTBASE, CARETUPBASE, CARETDOWNBASE) = xrange(12)
+ CARETLEFTBASE, CARETRIGHTBASE, CARETUPBASE, CARETDOWNBASE) = range(12)
 
 _empty_path = Path(np.empty((0, 2)))
 
diff --git a/lib/matplotlib/mathtext.py b/lib/matplotlib/mathtext.py
@@ -2641,7 +2641,7 @@ def symbol(self, s, loc, toks):
         if c in self._spaced_symbols:
             # iterate until we find previous character, needed for cases
             # such as ${ -2}$, $ -2$, or $   -2$.
-            for i in six.moves.xrange(1, loc + 1):
+            for i in range(1, loc + 1):
                 prev_char = s[loc-i]
                 if prev_char != ' ':
                     break
@@ -2660,11 +2660,11 @@ def symbol(self, s, loc, toks):
             # Do not space commas between brackets
             if c == ',':
                 prev_char, next_char = '', ''
-                for i in six.moves.xrange(1, loc + 1):
+                for i in range(1, loc + 1):
                     prev_char = s[loc - i]
                     if prev_char != ' ':
                         break
-                for i in six.moves.xrange(1, len(s) - loc):
+                for i in range(1, len(s) - loc):
                     next_char = s[loc + i]
                     if next_char != ' ':
                         break
diff --git a/lib/matplotlib/mlab.py b/lib/matplotlib/mlab.py
@@ -166,7 +166,7 @@
                         unicode_literals)
 
 import six
-from six.moves import map, xrange, zip
+from six.moves import map, zip
 
 import copy
 import csv
@@ -1453,7 +1453,7 @@ def cohere_pairs(X, ij, NFFT=256, Fs=2, detrend=detrend_none,
         windowVals = window
     else:
         windowVals = window(np.ones(NFFT, X.dtype))
-    ind = list(xrange(0, numRows-NFFT+1, NFFT-noverlap))
+    ind = list(range(0, numRows-NFFT+1, NFFT-noverlap))
     numSlices = len(ind)
     FFTSlices = {}
     FFTConjSlices = {}
diff --git a/lib/matplotlib/offsetbox.py b/lib/matplotlib/offsetbox.py
@@ -18,7 +18,7 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange, zip
+from six.moves import zip
 
 import warnings
 import matplotlib.transforms as mtransforms
@@ -1197,7 +1197,7 @@ def _get_anchored_bbox(self, loc, bbox, parentbbox, borderpad):
         """
         assert loc in range(1, 11)  # called only internally
 
-        BEST, UR, UL, LL, LR, R, CL, CR, LC, UC, C = xrange(11)
+        BEST, UR, UL, LL, LR, R, CL, CR, LC, UC, C = range(11)
 
         anchor_coefs = {UR: "NE",
                         UL: "NW",
diff --git a/lib/matplotlib/sphinxext/plot_directive.py b/lib/matplotlib/sphinxext/plot_directive.py
@@ -138,7 +138,6 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange
 
 import sys, os, shutil, io, re, textwrap
 from os.path import relpath
@@ -597,7 +596,7 @@ def render_figures(code, code_path, output_dir, output_base, context,
     all_exists = True
     for i, code_piece in enumerate(code_pieces):
         images = []
-        for j in xrange(1000):
+        for j in range(1000):
             if len(code_pieces) > 1:
                 img = ImageFile('%s_%02d_%02d' % (output_base, i, j),
                                 output_dir)
diff --git a/lib/matplotlib/stackplot.py b/lib/matplotlib/stackplot.py
@@ -9,9 +9,6 @@
 from __future__ import (absolute_import, division, print_function,
                         unicode_literals)
 
-import six
-from six.moves import xrange
-
 from cycler import cycler
 import numpy as np
 
@@ -120,7 +117,7 @@ def stackplot(axes, x, *args, **kwargs):
     r = [coll]
 
     # Color between array i-1 and array i
-    for i in xrange(len(y) - 1):
+    for i in range(len(y) - 1):
         color = axes._get_lines.get_next_color()
         r.append(axes.fill_between(x, stack[i, :], stack[i + 1, :],
                                    facecolor=color, label=next(labels, None),
diff --git a/lib/matplotlib/streamplot.py b/lib/matplotlib/streamplot.py
@@ -6,7 +6,6 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange
 
 import numpy as np
 import matplotlib
@@ -648,7 +647,7 @@ def _gen_starting_points(shape):
     x, y = 0, 0
     i = 0
     direction = 'right'
-    for i in xrange(nx * ny):
+    for i in range(nx * ny):
 
         yield x, y
 
diff --git a/lib/matplotlib/table.py b/lib/matplotlib/table.py
@@ -23,7 +23,6 @@
                         unicode_literals)
 
 import six
-from six.moves import xrange
 
 import warnings
 
@@ -551,7 +550,7 @@ def _update_positions(self, renderer):
         else:
             # Position using loc
             (BEST, UR, UL, LL, LR, CL, CR, LC, UC, C,
-             TR, TL, BL, BR, R, L, T, B) = xrange(len(self.codes))
+             TR, TL, BL, BR, R, L, T, B) = range(len(self.codes))
             # defaults for center
             ox = (0.5 - w / 2) - l
             oy = (0.5 - h / 2) - b
@@ -670,24 +669,24 @@ def table(ax,
     height = table._approx_text_height()
 
     # Add the cells
-    for row in xrange(rows):
-        for col in xrange(cols):
+    for row in range(rows):
+        for col in range(cols):
             table.add_cell(row + offset, col,
                            width=colWidths[col], height=height,
                            text=cellText[row][col],
                            facecolor=cellColours[row][col],
                            loc=cellLoc)
     # Do column labels
     if colLabels is not None:
-        for col in xrange(cols):
+        for col in range(cols):
             table.add_cell(0, col,
                            width=colWidths[col], height=height,
                            text=colLabels[col], facecolor=colColours[col],
                            loc=colLoc)
 
     # Do row labels
     if rowLabels is not None:
-        for row in xrange(rows):
+        for row in range(rows):
             table.add_cell(row + offset, -1,
                            width=rowLabelWidth or 1e-15, height=height,
                            text=rowLabels[row], facecolor=rowColours[row],
diff --git a/lib/matplotlib/testing/jpl_units/StrConverter.py b/lib/matplotlib/testing/jpl_units/StrConverter.py
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
diff --git a/lib/matplotlib/tri/triinterpolate.py b/lib/matplotlib/tri/triinterpolate.py
diff --git a/lib/mpl_toolkits/axes_grid1/colorbar.py b/lib/mpl_toolkits/axes_grid1/colorbar.py
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
