Raise a more helpful TypeError when trying to subscript a LazyFrame. (#…

…3554)
pola-rs · Jun 1, 2022 · 91d02cd · 91d02cd
1 parent 5b23e4e
commit 91d02cd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/py-polars/polars/internals/lazy_frame.py b/py-polars/polars/internals/lazy_frame.py
@@ -314,6 +314,11 @@ def _scan_python_function(cls, schema: "pa.schema", scan_fn: bytes) -> "LazyFram
         self._ldf = PyLazyFrame.scan_from_python_function(list(schema), scan_fn)
         return self
 
+    def __getitem__(self, item: Union[int, range, slice]) -> None:
+        raise TypeError(
+            "'LazyFrame' object is not subscriptable. Use 'select()' or 'filter()' instead."
+        )
+
     def pipe(self, func: Callable[..., Any], *args: Any, **kwargs: Any) -> Any:
         """
         Apply a function on Self.