PERF: custom ops for RangeIndex.[all|any|__contain__]

Author: topper-123

pandas/core/common.py

@@ -490,3 +490,13 @@ def f(x):
         f = mapper
 
     return f
+
+
+def ensure_python_int(value: Union[int, Any]) -> int:
+    msg = "Wrong type {} for value {}"
+    try:
+        new_value = int(value)
+        assert (new_value == value)
+    except (TypeError, ValueError, AssertionError):
+        raise TypeError(msg.format(type(value), value))
+    return new_value

pandas/core/indexes/base.py

@@ -4013,11 +4013,7 @@ def __contains__(self, key):
 
     @Appender(_index_shared_docs['contains'] % _index_doc_kwargs)
     def contains(self, key):
-        hash(key)
-        try:
-            return key in self._engine
-        except (TypeError, ValueError):
-            return False
+        return key in self
 
     def __hash__(self):
         raise TypeError("unhashable type: %r" % type(self).__name__)

pandas/core/indexes/range.py

@@ -334,6 +334,14 @@ def is_monotonic_decreasing(self):
     def has_duplicates(self):
         return False
 
+    def __contains__(self, key):
+        hash(key)
+        try:
+            key = com.ensure_python_int(key)
+        except TypeError:
+            return False
+        return key in self._range
+
     @Appender(_index_shared_docs['get_loc'])
     def get_loc(self, key, method=None, tolerance=None):
         if is_integer(key) and method is None and tolerance is None:
@@ -640,6 +648,14 @@ def __floordiv__(self, other):
                 return self._simple_new(start, start + 1, 1, name=self.name)
         return self._int64index // other
 
+    def all(self) -> bool:
+        if 0 in self._range:
+            return False
+        return True
+
+    def any(self) -> bool:
+        return any(self._range)
+
     @classmethod
     def _add_numeric_methods_binary(cls):
         """ add in numeric methods, specialized to RangeIndex """
@@ -725,4 +741,3 @@ def _evaluate_numeric_binop(self, other):
 
 
 RangeIndex._add_numeric_methods()
-RangeIndex._add_logical_methods()

pandas/tests/indexes/test_range.py

@@ -245,10 +245,9 @@ def test_dtype(self):
         assert self.index.dtype == np.int64
 
     def test_cached_data(self):
-        # GH 26565
-        # Calling RangeIndex._data caches an int64 array of the same length as
-        # self at self._cached_data.
-        # This tests whether _cached_data is being set by various operations.
+        # GH 26565, GH26617
+        # Calling RangeIndex._data caches an int64 array of the same length at
+        # self._cached_data. This tests whether _cached_data has been set.
         idx = RangeIndex(0, 100, 10)
 
         assert idx._cached_data is None
@@ -262,6 +261,24 @@ def test_cached_data(self):
         idx.get_loc(20)
         assert idx._cached_data is None
 
+        90 in idx
+        assert idx._cached_data is None
+
+        91 in idx
+        assert idx._cached_data is None
+
+        idx.contains(90)
+        assert idx._cached_data is None
+
+        idx.contains(91)
+        assert idx._cached_data is None
+
+        idx.all()
+        assert idx._cached_data is None
+
+        idx.any()
+        assert idx._cached_data is None
+
         df = pd.DataFrame({'a': range(10)}, index=idx)
 
         df.loc[50]