Function rv::misc::argmax

pub fn argmax<T: PartialOrd>(xs: &[T]) -> Vec<usize>

Indices of the largest element(s) in xs.

If there is more than one largest element, argmax returns the indices of all replicates.

Examples

use rv::misc::argmax;

let xs: Vec<u8> = vec![1, 2, 3, 4, 5, 4, 5];
let ys: Vec<u8> = vec![1, 2, 3, 4, 5, 4, 0];

assert_eq!(argmax(&xs), vec![4, 6]);
assert_eq!(argmax(&ys), vec![4]);