我如何将两个Rust向量通过3块插入到一个新向量中？

zaq34kh6 于 2022-11-24 发布在其他

关注(0)|答案(2)|浏览(138)

我需要一个惯用的方法来交织这两个向量：

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];

我使用了itertools chunks方法，但我不认为这是最好的实现，因为有两个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>>();

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

rust

来源：https://stackoverflow.com/questions/59322812/how-do-i-interleave-two-rust-vectors-by-chunks-of-threes-into-a-new-vector

2条答案

按热度按时间

bvhaajcl1#

似乎仅使用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一样。

赞(0）回复(0）举报 2022-11-24

r1zk6ea12#

您需要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);
}

赞(0）回复(0）举报 2022-11-24

我来回答

我如何将两个Rust向量通过3块插入到一个新向量中？

2条答案

相关问题

热门标签

最新问答