当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Rust Iterator.map_while用法及代码示例


本文简要介绍rust语言中 std::iter::Iterator.map_while 的用法。

用法

fn map_while<B, P>(self, predicate: P) -> MapWhile<Self, P> where    P: FnMut(Self::Item) -> Option<B>,

创建一个迭代器,它都基于谓词和映射产生元素。

map_while() 将闭包作为参数。它将在迭代器的每个元素上调用此闭包,并在返回 Some(_) 时产生元素。

例子

基本用法:

let a = [-1i32, 4, 0, 1];

let mut iter = a.iter().map_while(|x| 16i32.checked_div(*x));

assert_eq!(iter.next(), Some(-16));
assert_eq!(iter.next(), Some(4));
assert_eq!(iter.next(), None);

这是相同的示例,但使用 take_while map

let a = [-1i32, 4, 0, 1];

let mut iter = a.iter()
                .map(|x| 16i32.checked_div(*x))
                .take_while(|x| x.is_some())
                .map(|x| x.unwrap());

assert_eq!(iter.next(), Some(-16));
assert_eq!(iter.next(), Some(4));
assert_eq!(iter.next(), None);

在初始 None 后停止:

use std::convert::TryFrom;

let a = [0, 1, 2, -3, 4, 5, -6];

let iter = a.iter().map_while(|x| u32::try_from(*x).ok());
let vec = iter.collect::<Vec<_>>();

// We have more elements which could fit in u32 (4, 5), but `map_while` returned `None` for `-3`
// (as the `predicate` returned `None`) and `collect` stops at the first `None` encountered.
assert_eq!(vec, vec![0, 1, 2]);

因为map_while() 需要查看该值以查看是否应包含它,所以使用迭代器将看到它已被删除:

use std::convert::TryFrom;

let a = [1, 2, -3, 4];
let mut iter = a.iter();

let result: Vec<u32> = iter.by_ref()
                           .map_while(|n| u32::try_from(*n).ok())
                           .collect();

assert_eq!(result, &[1, 2]);

let result: Vec<i32> = iter.cloned().collect();

assert_eq!(result, &[4]);

-3 不再存在,因为它被消耗是为了查看迭代是否应该停止,但没有放回迭代器中。

请注意,与std::iter::Iterator.take_while这个迭代器是不是融合了。也没有指定该迭代器在第一个迭代器之后返回什么None被返回。如果您需要融合迭代器,请使用std::iter::Iterator.fuse.

相关用法


注:本文由纯净天空筛选整理自rust-lang.org大神的英文原创作品 std::iter::Iterator.map_while。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。