pub fn argmax<T: PartialOrd>(xs: &[T]) -> Vec<usize>
Expand description
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]);