polars.Expr.arg_max

Expr.arg_max() → Expr

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

Если максимальное значение встречается несколько раз, эта функция может произвольно вернуть индекс любого из таких значений. В этом случае не гарантируется, что при нескольких запусках будет возвращён один и тот же индекс.

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

Примеры

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