当前位置：首页 > article >正文

【Rust Iterator 之 fold,map,filter,for_each】

article 2024/11/27 1:11:35

Rust Iterator 之 fold,map,filter,for_each

前言
map
for_each
- 通过源码看for_each
fold
filter
总结

前言

在Iterator 一文中，我们提到过Iterator时惰性的，也就是当我们将容器转换成迭代器时不会产生任何的迭代行为，所以在使用时开发者还需要将迭代器与消费者适配器共同使用。

map

比较常见的例子是map方法，map在使用时如果开发者不使用消费者适配器，则run时不会发生任何的事情并且在编译期间会收到报错。
代码示例：

fn usemapmethod() {
    let mut v = vec![1, 2, 3, 5];
    let vv = v.iter().map(|x| *x + 1);
    println!("{:?}", vv);
}
fn main() {
    usemapmethod();
}

上述代码使用了map想要使用闭包将每一个元素加1，但是没有使用消费者适配器，造成的结果就是打印的结果将与v中数据完全一致，因为map并没有执行，如果想要让其正确执行，需要配合消费者适配器，如下：

fn usemapmethod() {
    let mut v = vec![1, 2, 3, 5];
    let vv: Vec<_> = v.iter().map(|x| *x + 1).collect();
    println!("{:?}", vv);
}
fn main() {
    usemapmethod();
}

for_each

for_each 如字面意思，遍历迭代器，处理数据，与map不同的是，它不需要消费者适配器，同时与map的返回值不同，它只返回（），上代码：

use std::collections::HashMap;
fn usemapmethod() {
    let v = vec![1, 2, 3, 5];
    let mut mymap = HashMap::new();
    let _ = v.iter().for_each(|x| {
        mymap.insert(*x, *x + 1);
    });
    println!("{:?}", mymap);
}
fn main() {
    usemapmethod();
}

可以看到，在处理时我们直接将返回值丢弃，因为不同于消费者适配器返回的结果，我们通常不会接收for_each的返回结果，因为通常没有什么意义。

通过源码看for_each


            // We override the default implementation, which uses `try_fold`,
            // because this simple implementation generates less LLVM IR and is
            // faster to compile.
            #[inline]
            fn for_each<F>(mut self, mut f: F)
            where
                Self: Sized,
                F: FnMut(Self::Item),
            {
                while let Some(x) = self.next() {
                    f(x);
                }
            }

可以看到，不同于map，在for_each 内部通过一个while循环直接进行了遍历。

fold

字面意思折叠，通过命名大概能猜测出fold的具体用途，既将通过闭包处理的数据最后叠加到一起。

use std::collections::HashMap;
fn usemapmethod() {
    let v = vec![1, 2, 3, 4, 6, 5];
    let result = v
        .iter()
        .fold(0, |acc, num| if num % 3 == 0 { acc + num } else { acc });

    println!("{}", result)
}
fn main() {
    usemapmethod();
}

filter

按照特定的条件过滤符合的原有数据，当我们使用fold传入的acc值是0的时候，通常我们可以通过filter+sum的方式实现相同的逻辑，那么具体在实际使用过程中，应该如何挑选这里并没有什么约定俗成，选择自己适合并且容易理解的方式就选择filter+sum这通常满足我们的简单过滤和累计，fold试用于相对更复杂一点的逻辑。( 个人理解，当然你可以使用不同的组合方式创造出适合自己业务逻辑的代码)

fn usemapmethod() {
    let v = vec![1, 2, 3, 4, 6, 5];
    let vv: i32 = v.iter().filter(|n| *n % 3 == 0).sum();

    println!("{}", vv)
}
fn main() {
    usemapmethod();
}