在 Rust 中获取 slice 或 Vec 中最大或最小浮点值的索引的惯用方法是什么

发布于12月23日

Assumption--Vec<f32>有任何NaN个值或表现出任何NaN个行为.

以以下样本集为例:

要获得上述列表中最高值的index(值可能为负值)，最简洁、最稳定的方法是什么？

我最初的try 大致如下:

let _tmp = *nets.iter().max_by(|i, j| i.partial_cmp(j).unwrap()).unwrap();    
let _i = nets.iter().position(|&element| element == _tmp).unwrap();

其中nets是&Vec<f32>.在我看来，这显然是不正确的.

与之类似的Python版本(考虑到上述假设):

_i = nets.index(max(nets))

推荐答案

我可能会这样做:

fn main() -> Result<(), Box<std::error::Error>> {
    let samples = vec![
        0.28, 0.3102, 0.9856, 0.3679, 0.3697, 0.46, 0.4311, 0.9781, 0.9891, 0.5052, 0.9173, 0.932,
        0.8365, 0.5822, 0.9981, 0.9977,
    ];

    // Use enumerate to get the index
    let mut iter = samples.iter().enumerate();
    // we get the first entry
    let init = iter.next().ok_or("Need at least one input")?;
    // we process the rest
    let result = iter.try_fold(init, |acc, x| {
        // return None if x is NaN
        let cmp = x.1.partial_cmp(acc.1)?;
        // if x is greater the acc
        let max = if let std::cmp::Ordering::Greater = cmp {
            x
        } else {
            acc
        };
        Some(max)
    });
    println!("{:?}", result);

    Ok(())
}

这可以通过在迭代器上添加一个trait来实现，例如函数try_max_by.