polars.Expr.top_k

Expr.top_k(
    k: int | IntoExprColumn = 5,
) → Expr

Возвращает k наибольших элементов.

Ненулевые элементы всегда имеют приоритет над нулевыми. Порядок элементов в результате не гарантируется. Если требуется отсортировать результат, вызовите sort() после этой функции.

Вычислительная сложность:

\[O(n)\]

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

Параметры:

k

Количество возвращаемых элементов.

См. также

top_k_by

bottom_k

bottom_k_by

Примеры

Получить 5 наибольших значений ряда.

>>> df = pl.DataFrame({"value": [1, 98, 2, 3, 99, 4]})
>>> df.select(
...     pl.col("value").top_k().alias("top_k"),
...     pl.col("value").bottom_k().alias("bottom_k"),
... )
shape: (5, 2)
┌───────┬──────────┐
│ top_k ┆ bottom_k │
│ ---   ┆ ---      │
│ i64   ┆ i64      │
╞═══════╪══════════╡
│ 4     ┆ 1        │
│ 98    ┆ 98       │
│ 2     ┆ 2        │
│ 3     ┆ 3        │
│ 99    ┆ 4        │
└───────┴──────────┘