Skip to content

Commit 261f43a

Browse files
committed
PERF: do not instantiate IndexEngine for standard lookup over RangeIndex
closes #16685
1 parent 53a36a7 commit 261f43a

File tree

3 files changed

+53
-2
lines changed

3 files changed

+53
-2
lines changed

doc/source/whatsnew/v0.25.0.rst

+1
Original file line numberDiff line numberDiff line change
@@ -628,6 +628,7 @@ Performance improvements
628628
int8/int16/int32 and the searched key is within the integer bounds for the dtype (:issue:`22034`)
629629
- Improved performance of :meth:`pandas.core.groupby.GroupBy.quantile` (:issue:`20405`)
630630
- Improved performance of slicing and other selected operation on a :class:`RangeIndex` (:issue:`26565`, :issue:`26617`, :issue:`26722`)
631+
- :class:`RangeIndex` now performs standard lookup without instantiating an actual hashtable, hence saving memory (:issue:`16685`)
631632
- Improved performance of :meth:`read_csv` by faster tokenizing and faster parsing of small float numbers (:issue:`25784`)
632633
- Improved performance of :meth:`read_csv` by faster parsing of N/A and boolean values (:issue:`25804`)
633634
- Improved performance of :attr:`IntervalIndex.is_monotonic`, :attr:`IntervalIndex.is_monotonic_increasing` and :attr:`IntervalIndex.is_monotonic_decreasing` by removing conversion to :class:`MultiIndex` (:issue:`24813`)

pandas/core/indexes/range.py

+32-2
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@
1313

1414
from pandas.core.dtypes import concat as _concat
1515
from pandas.core.dtypes.common import (
16-
ensure_python_int, is_int64_dtype, is_integer, is_scalar,
17-
is_timedelta64_dtype)
16+
ensure_python_int, is_int64_dtype, is_integer, is_scalar, is_integer_dtype,
17+
is_timedelta64_dtype, is_list_like, ensure_platform_int)
1818
from pandas.core.dtypes.generic import (
1919
ABCDataFrame, ABCSeries, ABCTimedeltaIndex)
2020

@@ -348,6 +348,36 @@ def get_loc(self, key, method=None, tolerance=None):
348348
raise KeyError(key)
349349
return super().get_loc(key, method=method, tolerance=tolerance)
350350

351+
@Appender(_index_shared_docs['get_indexer'])
352+
def get_indexer(self, target, method=None, limit=None, tolerance=None):
353+
if not (method is None and tolerance is None and is_list_like(target)):
354+
return super().get_indexer(target, method=method,
355+
tolerance=tolerance)
356+
357+
if self.step > 0:
358+
start, stop, step = self.start, self.stop, self.step
359+
else:
360+
# Work on reversed range for simplicity:
361+
start, stop, step = (self.stop - self.step,
362+
self.start + 1,
363+
- self.step)
364+
365+
target_array = np.asarray(target)
366+
if not (is_integer_dtype(target_array) and target_array.ndim == 1):
367+
# checks/conversions/roundings are delegated to general method
368+
return super().get_indexer(target, method=method,
369+
tolerance=tolerance)
370+
371+
locs = target_array - start
372+
valid = (locs % step == 0) & (locs >= 0) & (target_array < stop)
373+
locs[~valid] = -1
374+
locs[valid] = locs[valid] / step
375+
376+
if step != self.step:
377+
# We reversed this range: transform to original locs
378+
locs[valid] = len(self) - 1 - locs[valid]
379+
return ensure_platform_int(locs)
380+
351381
def tolist(self):
352382
return list(self._range)
353383

pandas/tests/indexes/test_range.py

+20
Original file line numberDiff line numberDiff line change
@@ -965,3 +965,23 @@ def test_append(self, appends):
965965
# Append single item rather than list
966966
result2 = indices[0].append(indices[1])
967967
tm.assert_index_equal(result2, expected, exact=True)
968+
969+
def test_engineless_lookup(self):
970+
# GH 16685
971+
# Standard lookup on RangeIndex should not require the engine to be
972+
# created
973+
idx = RangeIndex(2, 10, 3)
974+
975+
assert idx.get_loc(5) == 1
976+
tm.assert_numpy_array_equal(idx.get_indexer([2, 8]),
977+
np.array([0, 2]))
978+
with pytest.raises(KeyError):
979+
idx.get_loc(3)
980+
981+
assert '_engine' not in idx._cache
982+
983+
# The engine is still required for lookup of a different dtype scalar:
984+
with pytest.raises(KeyError):
985+
assert idx.get_loc('a') == -1
986+
987+
assert '_engine' in idx._cache

0 commit comments

Comments
 (0)