获取向量中满足条件的所有索引

Question

我想知道是否可以获取满足 Rust 向量数据类型中的条件的所有索引。我知道特征运算符提供了一种方法来查找执行它的第一个元素：

let test=vec![1,0,0,1,1];
let index = test.iter().position(|&r| r == 1);
println!("{}",index) //0

但是，我有兴趣在测试 vec 中获取所有等于 1 的索引。

let test=vec![1,0,0,1,1];
let indexes =  //Some Code
println!("{:?}",indexes) //0,3,4

我该怎么办？

Answer 1

使用enumerate():

let test = vec![1, 0, 0, 1, 1];
let indices = test
    .iter()
    .enumerate()
    .filter(|(_, &r)| r == 1)
    .map(|(index, _)| index)
    .collect::<Vec<_>>();
dbg!(indices); // 0, 3, 4

Playground.

你也可以结合filter()和map()，虽然我认为前一个版本更干净：

let indices = test
    .iter()
    .enumerate()
    .filter_map(|(index, &r)| if r == 1 { Some(index) } else { None })
    .collect::<Vec<_>>();

或

let indices = test
    .iter()
    .enumerate()
    .filter_map(|(index, &r)| (r == 1).then(|| index))
    .collect::<Vec<_>>();

获取向量中满足条件的所有索引

Get all the indexes that fulfill a condition in a vector

vector

rust