Python rename drop_column(s) to drop (union eager/lazy) and pl.exclud…

…e syntactic sugar

py-polars/docs/source/reference/expression.rst

@@ -42,6 +42,7 @@ These functions can be used as expression and sometimes also in eager contexts.
    concat_str
    concat_list
    when
+   exclude
 
 Constructor
 -----------

py-polars/docs/source/reference/lazyframe.rst

@@ -38,9 +38,8 @@ Manipulation/ selection
     LazyFrame.join
     LazyFrame.with_columns
     LazyFrame.with_column
-    LazyFrame.drop_columns
-    LazyFrame.drop_column
     LazyFrame.with_column_renamed
+    LazyFrame.drop
     LazyFrame.rename
     LazyFrame.reverse
     LazyFrame.shift

py-polars/polars/lazy/frame.py

@@ -548,27 +548,20 @@ def with_column(self, expr: "Expr") -> "LazyFrame":
         """
         return self.with_columns([expr])
 
-    def drop_columns(self, columns: tp.List[str]) -> "LazyFrame":
+    def drop(self, columns: Union[str, tp.List[str]]) -> "LazyFrame":
         """
-        Remove multiple columns from a DataFrame.
+        Remove one or multiple columns from a DataFrame.
 
         Parameters
         ----------
         columns
-            List of column names.
-        """
-        return wrap_ldf(self._ldf.drop_columns(columns))
+            - Name of the column that should be removed.
+            - List of column names.
 
-    def drop_column(self, column: str) -> "LazyFrame":
         """
-        Remove a column from the DataFrame.
-
-        Parameters
-        ----------
-        column
-            Name of the column that should be removed.
-        """
-        return self.drop_columns([column])
+        if isinstance(columns, str):
+            columns = [columns]
+        return wrap_ldf(self._ldf.drop_columns(columns))
 
     def with_column_renamed(self, existing_name: str, new_name: str) -> "LazyFrame":
         """

py-polars/polars/lazy/functions.py

@@ -517,6 +517,56 @@ def any(name: Union[str, tp.List["pl.Expr"]]) -> "pl.Expr":
     return col(name).sum() > 0
 
 
+def exclude(columns: Union[str, tp.List[str]]) -> "pl.Expr":
+    """
+     Exclude certain columns from a wildcard expression.
+
+     Syntactic sugar for:
+     >>> col("*").exclude()
+
+     Parameters
+     ----------
+     columns
+         Column(s) to exclude from selection
+
+     Examples
+     --------
+
+     >>> df = pl.DataFrame({
+     >>>     "a": [1, 2, 3],
+     >>>     "b": ["a", "b", None],
+     >>>     "c": [None, 2, 1]
+     >>> })
+     >>> df
+     shape: (3, 3)
+     ╭─────┬──────┬──────╮
+     │ a   ┆ b    ┆ c    │
+     │ --- ┆ ---  ┆ ---  │
+     │ i64 ┆ str  ┆ i64  │
+     ╞═════╪══════╪══════╡
+     │ 1   ┆ "a"  ┆ null │
+     ├╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
+     │ 2   ┆ "b"  ┆ 2    │
+     ├╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
+     │ 3   ┆ null ┆ 1    │
+     ╰─────┴──────┴──────╯
+     >>> df.select(pl.exclude("b"))
+    shape: (3, 2)
+     ╭─────┬──────╮
+     │ a   ┆ c    │
+     │ --- ┆ ---  │
+     │ i64 ┆ i64  │
+     ╞═════╪══════╡
+     │ 1   ┆ null │
+     ├╌╌╌╌╌┼╌╌╌╌╌╌┤
+     │ 2   ┆ 2    │
+     ├╌╌╌╌╌┼╌╌╌╌╌╌┤
+     │ 3   ┆ 1    │
+     ╰─────┴──────╯
+    """
+    return col("*").exclude(columns)
+
+
 def all(name: Optional[Union[str, tp.List["pl.Expr"]]] = None) -> "pl.Expr":
     """
     This function is two things

py-polars/tests/test_lazy.py

@@ -581,7 +581,7 @@ def test_rename():
 
 
 def test_drop_columns():
-    out = pl.DataFrame({"a": [1], "b": [2], "c": [3]}).lazy().drop_columns(["a", "b"])
+    out = pl.DataFrame({"a": [1], "b": [2], "c": [3]}).lazy().drop(["a", "b"])
     assert out.columns == ["c"]