如何将两个 Rust 向量交错成一个三块到一个新的向量中?

Question

我需要一种惯用的方式来交错这两个向量：

let v1 = vec![1.0, 2.0, 3.0, 4.0, 5.0, 6.0];
let v2 = vec![7.0, 8.0, 9.0, 10.0, 11.0, 12.0];

我期望的输出是：

[1.0, 2.0, 3.0,
7.0, 8.0, 9.0,
4.0, 5.0, 6.0,
10.0, 11.0, 12.0];

我使用了 itertoolschunks方法，但我不认为这是最好的实现，因为有两个collect调用。

let output = interleave(&v1.into_iter().chunks(3), &v2.into_iter().chunks(3)).map(|v| {v.into_iter().collect::<Vec<f32>>()}).flatten().collect::<Vec<f32>>();

有没有更好的方法来使用迭代工具编写这个迭代器？

Answer 1

似乎只有 std 就可以实现相同的效果：

fn main() {
let v1 = vec![1.0, 2.0, 3.0, 4.0, 5.0, 6.0];
let v2 = vec![7.0, 8.0, 9.0, 10.0, 11.0, 12.0];
let v3: Vec<f64> = v1.chunks(3)
.zip(v2.chunks(3)) // yields items like (&[1.0, 2.0, 3.0], &[7.0, 8.0, 9.0])
.flat_map(|(a, b)| a.into_iter().chain(b)) // chains to produce iterators like [1.0, 2.0, 3.0, 7.0, 8.0, 9.0]
.copied() // &f64 -> f64, optional
.collect();
println!("v3 is {:?}", v3);
}

即zip -> chain表现得像interleave.

Answer 2

你想要Iterator::flatten：

use itertools::interleave;
fn main() {
let v1 = vec![1.0, 2.0, 3.0, 4.0, 5.0, 6.0];
let v2 = vec![7.0, 8.0, 9.0, 10.0, 11.0, 12.0];
let v = interleave(v1.chunks(3), v2.chunks(3))
.flatten()
.collect::<Vec<_>>();
dbg!(v);
}