rust - 获取分区函数中项目的索引

Question

我需要在 Rust 中拆分数组/向量并找到 partition在文档中。看起来传递给 partition 的回调函数只能访问数组的项目。我如何获得项目的索引？

例如，给定一个数组[1, 2, 3, 4]，我如何根据它们的位置将它分成两个，所以第一个是[1, 3 ] 因为它们每个都有一个偶数位置(1 和 3 的索引为 0 和 2，这是一个偶数)，第二个将是 [2, 4]。

Answer 1

解决方案可能是使用 enumerate和 partition_map来自 itertools:

use itertools::{Either, Itertools};

fn main() {
    let a = vec![1, 2, 3, 4];

    let (b, c): (Vec<_>, Vec<_>) = a.into_iter().enumerate().partition_map(|(i, foo)| {
        if i % 2 == 0 {
            Either::Left(foo)
        } else {
            Either::Right(foo)
        }
    });

    println!("{:?}, {:?}", b, c);
}