CLN: remove compat.lrange

Author: topper-123

pandas/compat/__init__.py

@@ -4,9 +4,6 @@
 
 Cross-compatible functions for different versions of Python.
 
-Key items to import for compatible code:
-* lists: lrange()
-
 Other items:
 * platform checker
 """
@@ -19,11 +16,6 @@
 PYPY = platform.python_implementation() == 'PyPy'
 
 
-# list-producing versions of the major Python iterating functions
-def lrange(*args, **kwargs):
-    return list(range(*args, **kwargs))
-
-
 # ----------------------------------------------------------------------------
 # functions largely based / taken from the six module
 

pandas/plotting/_converter.py

@@ -12,7 +12,6 @@
 from pandas._libs import lib, tslibs
 from pandas._libs.tslibs import resolution
 from pandas._libs.tslibs.frequencies import FreqGroup, get_freq
-from pandas.compat import lrange
 
 from pandas.core.dtypes.common import (
     is_datetime64_ns_dtype, is_float, is_float_dtype, is_integer,
@@ -1029,7 +1028,7 @@ def __call__(self):
             base = self.base
             (d, m) = divmod(vmin, base)
             vmin = (d + 1) * base
-            locs = lrange(vmin, vmax + 1, base)
+            locs = list(range(vmin, vmax + 1, base))
         return locs
 
     def autoscale(self):

pandas/plotting/_core.py

@@ -8,7 +8,6 @@
 
 from pandas._config import get_option
 
-from pandas.compat import lrange
 from pandas.errors import AbstractMethodError
 from pandas.util._decorators import Appender, cache_readonly
 
@@ -583,9 +582,9 @@ def _get_xticks(self, convert_period=False):
                 x = self.data.index._mpl_repr()
             else:
                 self._need_to_set_index = True
-                x = lrange(len(index))
+                x = list(range(len(index)))
         else:
-            x = lrange(len(index))
+            x = list(range(len(index)))
 
         return x
 

pandas/plotting/_misc.py

@@ -1,7 +1,6 @@
 # being a bit too dynamic
 import numpy as np
 
-from pandas.compat import lrange
 from pandas.util._decorators import deprecate_kwarg
 
 from pandas.core.dtypes.missing import notna
@@ -81,8 +80,8 @@ def scatter_matrix(frame, alpha=0.5, figsize=None, ax=None, grid=False,
         rdelta_ext = (rmax_ - rmin_) * range_padding / 2.
         boundaries_list.append((rmin_ - rdelta_ext, rmax_ + rdelta_ext))
 
-    for i, a in zip(lrange(n), df.columns):
-        for j, b in zip(lrange(n), df.columns):
+    for i, a in enumerate(df.columns):
+        for j, b in enumerate(df.columns):
             ax = axes[i, j]
 
             if i == j:
@@ -420,7 +419,7 @@ def bootstrap_plot(series, fig=None, size=50, samples=500, **kwds):
                           for sampling in samplings])
     if fig is None:
         fig = plt.figure()
-    x = lrange(samples)
+    x = list(range(samples))
     axes = []
     ax1 = fig.add_subplot(2, 3, 1)
     ax1.set_xlabel("Sample")
@@ -532,7 +531,7 @@ def parallel_coordinates(frame, class_column, cols=None, ax=None, color=None,
             raise ValueError('Length of xticks must match number of columns')
         x = xticks
     else:
-        x = lrange(ncols)
+        x = list(range(ncols))
 
     if ax is None:
         ax = plt.gca()

pandas/plotting/_style.py

@@ -4,8 +4,6 @@
 
 import numpy as np
 
-from pandas.compat import lrange
-
 from pandas.core.dtypes.common import is_list_like
 
 
@@ -49,7 +47,7 @@ def random_color(column):
                 rs = com.random_state(column)
                 return rs.rand(3).tolist()
 
-            colors = [random_color(num) for num in lrange(num_colors)]
+            colors = [random_color(num) for num in range(num_colors)]
         else:
             raise ValueError("color_type must be either 'default' or 'random'")
 

pandas/tests/arrays/categorical/test_missing.py

@@ -3,8 +3,6 @@
 import numpy as np
 import pytest
 
-from pandas.compat import lrange
-
 from pandas.core.dtypes.dtypes import CategoricalDtype
 
 from pandas import Categorical, Index, isna
@@ -16,7 +14,7 @@ class TestCategoricalMissing:
     def test_na_flags_int_categories(self):
         # #1457
 
-        categories = lrange(10)
+        categories = list(range(10))
         labels = np.random.randint(0, 10, 20)
         labels[::5] = -1
 

pandas/tests/frame/test_duplicates.py

@@ -1,8 +1,6 @@
 import numpy as np
 import pytest
 
-from pandas.compat import lrange
-
 from pandas import DataFrame, Series
 import pandas.util.testing as tm
 
@@ -88,8 +86,8 @@ def test_drop_duplicates():
                     'B': ['one', 'one', 'two', 'two',
                           'two', 'two', 'one', 'two'],
                     'C': [1, 1, 2, 2, 2, 2, 1, 2],
-                    'D': lrange(8)})
-
+                    'D': range(8),
+                    })
     # single column
     result = df.drop_duplicates('AAA')
     expected = df[:2]
@@ -211,8 +209,8 @@ def test_drop_duplicates_for_take_all():
                     'B': ['one', 'one', 'two', 'two',
                           'two', 'two', 'one', 'two'],
                     'C': [1, 1, 2, 2, 2, 2, 1, 2],
-                    'D': lrange(8)})
-
+                    'D': range(8),
+                    })
     # single column
     result = df.drop_duplicates('AAA')
     expected = df.iloc[[0, 1, 2, 6]]
@@ -246,8 +244,8 @@ def test_drop_duplicates_tuple():
                     'B': ['one', 'one', 'two', 'two',
                           'two', 'two', 'one', 'two'],
                     'C': [1, 1, 2, 2, 2, 2, 1, 2],
-                    'D': lrange(8)})
-
+                    'D': range(8),
+                    })
     # single column
     result = df.drop_duplicates(('AA', 'AB'))
     expected = df[:2]
@@ -292,8 +290,8 @@ def test_drop_duplicates_NA():
                     'B': ['one', 'one', 'two', 'two',
                           'two', 'two', 'one', 'two'],
                     'C': [1.0, np.nan, np.nan, np.nan, 1., 1., 1, 1.],
-                    'D': lrange(8)})
-
+                    'D': range(8),
+                    })
     # single column
     result = df.drop_duplicates('A')
     expected = df.loc[[0, 2, 3]]
@@ -327,8 +325,8 @@ def test_drop_duplicates_NA():
                     'B': ['one', 'one', 'two', 'two',
                           'two', 'two', 'one', 'two'],
                     'C': [1.0, np.nan, np.nan, np.nan, 1., 1., 1, 1.],
-                    'D': lrange(8)})
-
+                    'D': range(8),
+                    })
     # single column
     result = df.drop_duplicates('C')
     expected = df[:2]
@@ -398,8 +396,8 @@ def test_drop_duplicates_inplace():
                       'B': ['one', 'one', 'two', 'two',
                             'two', 'two', 'one', 'two'],
                       'C': [1, 1, 2, 2, 2, 2, 1, 2],
-                      'D': lrange(8)})
-
+                      'D': range(8),
+                      })
     # single column
     df = orig.copy()
     df.drop_duplicates('A', inplace=True)

pandas/tests/indexes/datetimes/test_timezones.py

@@ -11,7 +11,6 @@
 import pytz
 
 from pandas._libs.tslibs import conversion, timezones
-from pandas.compat import lrange
 import pandas.util._test_decorators as td
 
 import pandas as pd
@@ -958,7 +957,7 @@ def test_dti_with_timezone_repr(self, tzstr):
     def test_dti_take_dont_lose_meta(self, tzstr):
         rng = date_range('1/1/2000', periods=20, tz=tzstr)
 
-        result = rng.take(lrange(5))
+        result = rng.take(range(5))
         assert result.tz == rng.tz
         assert result.freq == rng.freq
 

pandas/tests/indexes/multi/test_analytics.py

@@ -1,7 +1,6 @@
 import numpy as np
 import pytest
 
-from pandas.compat import lrange
 from pandas.compat.numpy import _np_version_under1p17
 
 import pandas as pd
@@ -32,8 +31,8 @@ def test_groupby(idx):
 
 
 def test_truncate():
-    major_axis = Index(lrange(4))
-    minor_axis = Index(lrange(2))
+    major_axis = Index(list(range(4)))
+    minor_axis = Index(list(range(2)))
 
     major_codes = np.array([0, 0, 1, 2, 3, 3])
     minor_codes = np.array([0, 1, 0, 1, 0, 1])

pandas/tests/indexes/multi/test_constructor.py

@@ -4,7 +4,6 @@
 import pytest
 
 from pandas._libs.tslib import Timestamp
-from pandas.compat import lrange
 
 from pandas.core.dtypes.cast import construct_1d_object_array_from_listlike
 
@@ -383,7 +382,7 @@ def test_from_product_empty_two_levels(first, second):
 def test_from_product_empty_three_levels(N):
     # GH12258
     names = ['A', 'B', 'C']
-    lvl2 = lrange(N)
+    lvl2 = list(range(N))
     result = MultiIndex.from_product([[], lvl2, []], names=names)
     expected = MultiIndex(levels=[[], lvl2, []],
                           codes=[[], [], []], names=names)