Rust迭代器收集为新Vec：解决跨向量随机选元素类型报错

错误原因 #

你的代码中，xs.iter()和ys.iter()返回的是引用类型迭代器，每个元素的类型是&u8。经过map处理后，迭代器的输出项依然是&u8，但你需要收集成Vec<u8>（值类型的容器），类型不匹配导致报错。

修正方案 #

因为u8实现了Copy trait，有几种简单的方式解决类型不匹配问题：

方案1：直接解引用元素

在map闭包中对引用进行解引用，将&u8转为u8：

fn select(xs: &Vec<u8>, ys: &Vec<u8>, p: f32) -> Vec<u8> {
    let mut rng = rand::thread_rng();
    xs
        .iter()
        .zip(ys.iter())
        .map(|(a, b)| if rng.gen_range(0.0..1.0) < p { *a } else { *b })
        .collect()
}

方案2：使用copied()转换迭代器类型

在iter()之后调用copied()，直接把引用迭代器转为值类型迭代器：

fn select(xs: &Vec<u8>, ys: &Vec<u8>, p: f32) -> Vec<u8> {
    let mut rng = rand::thread_rng();
    xs
        .iter()
        .copied()
        .zip(ys.iter().copied())
        .map(|(a, b)| if rng.gen_range(0.0..1.0) < p { a } else { b })
        .collect()
}

方案3：使用cloned()转换迭代器类型

cloned()和copied()效果类似，但适用于所有实现了Clone trait的类型（u8也满足），通用性更强：

fn select(xs: &Vec<u8>, ys: &Vec<u8>, p: f32) -> Vec<u8> {
    let mut rng = rand::thread_rng();
    xs
        .iter()
        .cloned()
        .zip(ys.iter().cloned())
        .map(|(a, b)| if rng.gen_range(0.0..1.0) < p { a } else { b })
        .collect()
}

以上三种方案都能让迭代器输出u8类型的元素，从而正常收集为Vec<u8>。

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