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

@@ -315,6 +315,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:
@@ -673,6 +681,17 @@ def __floordiv__(self, other):
                     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:
+        rng = self._range
+        if len(rng) == 0 or (len(rng) == 1 and 0 in rng):
+            return False
+        return True
+
     @classmethod
     def _add_numeric_methods_binary(cls):
         """ add in numeric methods, specialized to RangeIndex """
@@ -761,4 +780,3 @@ def _evaluate_numeric_binop(self, other):
 
 
 RangeIndex._add_numeric_methods()
-RangeIndex._add_logical_methods()

pandas/tests/indexes/test_range.py

@@ -258,6 +258,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]
@@ -273,7 +291,7 @@ def test_cached_data(self):
         df.iloc[5:10]
         assert idx._cached_data is None
 
-        # actually calling data._data
+        # actually calling idx._data
         assert isinstance(idx._data, np.ndarray)
         assert isinstance(idx._cached_data, np.ndarray)