Bootstrap

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

Rust Iterator 之 fold,map,filter,for_each


前言

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();
}


总结

在实际开发中会遇到各种各样的需求,熟练掌握method并理解其中的意义和作用,可以让我们更加rusty。

“积跬步,行万里。”

;