如何在Rust中让map函数输出多个结果供后续处理？

要实现第一个映射步骤输出多个元素、再传递给第二个map处理的需求，你需要用**flat_map**代替第一个map。flat_map的作用是接收一个闭包，闭包返回一个迭代器，它会把所有这些迭代器的元素展开成一个连续的流，刚好满足一对多的映射需求。

修改后的代码示例如下：

let vector = [1, 2, 3];
let result: Vec<_> = vector.iter()
    .flat_map(|x| [x * 2, x * 3].into_iter()) // 每个输入元素生成2个结果
    .map(|x| x + 1) // 逐个处理每个生成的元素
    .collect();
println!("After mapping: {:?}", result);
// 输出结果: [3, 4, 5, 6, 7, 8]

补充说明： #

• 这里闭包返回的是一个数组转成的迭代器（[x*2, x*3].into_iter()），你也可以用std::iter::once配合chain来生成多元素迭代器，效果完全一致：

  .flat_map(|x| std::iter::once(x * 2).chain(std::iter::once(x * 3)))
  

• map只能实现一对一映射，因为它的闭包必须返回单个值；而flat_map通过返回迭代器的方式，让每个输入元素可以生成任意数量的输出元素，并且自动扁平化这些元素，让后续的map能逐个处理它们。

内容的提问来源于stack exchange，提问作者Winston