polars.LazyFrame.select

LazyFrame.select(
    *exprs: IntoExpr | Iterable[IntoExpr],
    **named_exprs: IntoExpr,
) → LazyFrame

Выбрать столбцы из этого LazyFrame.

движок:В памятиПотоковая обработкаРаспределённая обработка

Параметры:

*exprs

Столбец или столбцы для выбора, указанные в виде позиционных аргументов. Принимаются выражения. Строки интерпретируются как имена столбцов, остальные входные данные, не являющиеся выражениями, — как литералы.

**named_exprs

Дополнительные столбцы для выбора, указанные в виде именованных аргументов. Столбцам будут присвоены имена соответствующих аргументов.

Примеры

Передайте имя столбца, чтобы выбрать его.

>>> lf = pl.LazyFrame(
...     {
...         "foo": [1, 2, 3],
...         "bar": [6, 7, 8],
...         "ham": ["a", "b", "c"],
...     }
... )
>>> lf.select("foo").collect()
shape: (3, 1)
┌─────┐
│ foo │
│ --- │
│ i64 │
╞═════╡
│ 1   │
│ 2   │
│ 3   │
└─────┘

Несколько столбцов можно выбрать, передав список их имён.

>>> lf.select(["foo", "bar"]).collect()
shape: (3, 2)
┌─────┬─────┐
│ foo ┆ bar │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 1   ┆ 6   │
│ 2   ┆ 7   │
│ 3   ┆ 8   │
└─────┴─────┘

Несколько столбцов также можно выбрать с помощью позиционных аргументов вместо списка. Также принимаются выражения.

>>> lf.select(pl.col("foo"), pl.col("bar") + 1).collect()
shape: (3, 2)
┌─────┬─────┐
│ foo ┆ bar │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 1   ┆ 7   │
│ 2   ┆ 8   │
│ 3   ┆ 9   │
└─────┴─────┘

Используйте именованные аргументы, чтобы задать имена входным выражениям.

>>> lf.select(
...     threshold=pl.when(pl.col("foo") > 2).then(10).otherwise(0)
... ).collect()
shape: (3, 1)
┌───────────┐
│ threshold │
│ ---       │
│ i32       │
╞═══════════╡
│ 0         │
│ 0         │
│ 10        │
└───────────┘