polars.Expr.unique_counts

Expr.unique_counts() → Expr

Возвращает количество уникальных значений в порядке их появления.

Этот метод отличается от value_counts тем, что возвращает не сами значения, а только их количество, и может работать быстрее.

движок:В памятиПотоковый

Примеры

>>> df = pl.DataFrame(
...     {
...         "id": ["a", "b", "b", "c", "c", "c"],
...     }
... )
>>> df.select(pl.col("id").unique_counts())
shape: (3, 1)
┌─────┐
│ id  │
│ --- │
│ u32 │
╞═════╡
│ 1   │
│ 2   │
│ 3   │
└─────┘

Обратите внимание, что group_by можно использовать для вычисления количества.

>>> df.group_by("id", maintain_order=True).len().select("len")
shape: (3, 1)
┌─────┐
│ len │
│ --- │
│ u32 │
╞═════╡
│ 1   │
│ 2   │
│ 3   │
└─────┘

Чтобы добавить количество в качестве нового столбца, pl.len() можно использовать как оконную функцию.

>>> df.with_columns(pl.len().over("id"))
shape: (6, 2)
┌─────┬─────┐
│ id  ┆ len │
│ --- ┆ --- │
│ str ┆ u32 │
╞═════╪═════╡
│ a   ┆ 1   │
│ b   ┆ 2   │
│ b   ┆ 2   │
│ c   ┆ 3   │
│ c   ┆ 3   │
│ c   ┆ 3   │
└─────┴─────┘