时间戳

首页 > 解决方案 > Rust 迭代器类似于 chunks()

rust - Rust 迭代器类似于 chunks()

问题描述

这个迭代器

let data = vec![0, 1, 2, 3, 4, 5];

for x in data.chunks(2) {
    println!("{:?}", x);
}

会产生

[0, 1]
[2, 3]
[4, 5]

我可以使用迭代器来获得这样的东西吗？

[0, 1]
[1, 2]
[2, 3]
[3, 4]
[4, 5]

我知道如何使用 for 循环来做到这一点。但是迭代器可以做得更好吗？

标签： rustiterator

解决方案

我想您可以为此使用 Itertools.tuple_windows。根据文档，它“在所有连续窗口上返回一个迭代器，生成特定大小的元组（最多 4 个）”：

use itertools::Itertools;
use itertools::TupleWindows;
use std::slice::Iter;

let data = vec![0, 1, 2, 3, 4, 5];

let it: TupleWindows<Iter<'_, i32>, (&i32, &i32)> = data.iter().tuple_windows();

for elem in it {
    println!("{:?}", elem);
}

输出：

(0, 1)
(1, 2)
(2, 3)
(3, 4)
(4, 5)

编辑：正如@Masklinn的comment1 和@SebastianRedl的comment2 所述，您也可以windows从stdlib 使用并避免包含Itertools在您的项目中。但请注意，它仅适用于切片（或强制切片的事物），而不适用于一般迭代器（这在您的情况下很好）。

let data = vec![0, 1, 2, 3, 4, 5];

let it = data.windows(2);

for elem in it {
    println!("{:?}", elem);
}

输出：

[0, 1]
[1, 2]
[2, 3]
[3, 4]
[4, 5]

推荐阅读