polars.Series.arr.arg_max#

Series.arr.arg_max() → Series[source]#

Retrieve the index of the maximum value in every sub-array.

Returns:

Series: Series of data type UInt32 or UInt64 (depending on compilation).

Examples

>>> s = pl.Series("a", [[0, 9, 3], [9, 1, 2]], dtype=pl.Array(pl.Int64, 3))
>>> s.arr.arg_max()
shape: (2,)
Series: 'a' [u32]
[
    1
    0
]