polars.Expr.arg_sort

Expr.arg_sort(
    *,
    descending: bool = False,
    nulls_last: bool = False,
) → Expr

Получить значения индексов, которые отсортировали бы этот столбец.

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

Параметры:

descending

Сортировать по убыванию.

nulls_last

Размещать значения null в конце, а не в начале.

Возвращает:

Expr

Выражение типа данных UInt32.

См. также

Expr.gather

Получить значения по индексу.

Expr.rank

Получить ранг каждой строки.

Примеры

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

Используйте gather, чтобы применить arg_sort к другим столбцам.

>>> df.select(pl.col("b").gather(pl.col("a").arg_sort()))
shape: (3, 1)
┌─────┐
│ b   │
│ --- │
│ i64 │
╞═════╡
│ 2   │
│ 1   │
│ 3   │
└─────┘