polars.Series.arr.arg_max

Series.arr.arg_max() → Series

Retrieve the index of the maximum value in every sublist.

Returns:

Series of dtype UInt32/UInt64 (depending on compilation)