perf(python): speed up in series 10x (pola-rs#9794)

potzenhotz · Jul 11, 2023 · dbe7b51 · dbe7b51
1 parent 68f3f55
commit dbe7b51
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/py-polars/polars/series/series.py b/py-polars/polars/series/series.py
@@ -869,6 +869,13 @@ def __copy__(self) -> Self:
     def __deepcopy__(self, memo: None = None) -> Self:
         return self.clone()
 
+    def __contains__(self, item: Any) -> bool:
+        # TODO! optimize via `is_in` and `SORTED` flags
+        try:
+            return (self == item).any()
+        except ValueError:
+            return False
+
     def __iter__(self) -> Generator[Any, None, None]:
         if self.dtype == List:
             # TODO: either make a change and return py-native list data here, or find