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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。