如何将Rust中iter.take()转换为Vec<T>或&[T]？

定义了如下Cell结构体：

pub struct Cell {
    x: X, // 其他结构体
    y: Y, // 其他结构体
    weight: usize,
}

尝试从Row（Cell的集合）中选出优先级最高的单元格，编写了select_preference函数：

// 返回前n个权重为正的匹配单元格
pub fn select_preference(&mut self) -> Vec<Cell> {
    let top = 3;

    self.sort();
    // 排序后，过滤掉权重为0的单元格
    // 然后选择优先级最高的前几个单元格
    self.cells.split(|cell| cell.weight() == 0).take(top)
}

编译时出现类型不匹配错误：

Compiling playground v0.0.1 (/playground)
error[E0308]: mismatched types
  --> src/lib.rs:35:9
   |
29 |     pub fn select_preference(&mut self) -> Vec<Cell> {
   |                                            --------- 预期返回 `Vec<Cell>`
...
35 |         self.cells.split(|cell| cell.weight() == 0).take(top)
   |         ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 预期结构体 `Vec`，实际找到结构体 `std::iter::Take`
   |
   = note: expected struct `Vec<Cell>`
              found struct `std::iter::Take<std::slice::Split<'_, Cell, [closure@src/lib.rs:35:26: 35:32]>>`

已知Take是迭代器，但不清楚如何将其转换为Vec<Cell>或&[T]，寻求解决方法。

问题核心 #

1. split方法返回的是切片迭代器，每个元素是&[Cell]类型的切片，而非单个Cell。
2. take(top)得到的仍是迭代器，必须通过collect()方法将其转换为Vec。
3. split的行为是按权重为0的单元格分割切片，这和“过滤所有权重为0的元素”的需求不符——排序后0元素通常集中在末尾，更适合用过滤或取第一个非零片段。

修正方案 #

方案一：过滤+收集（推荐）

直接过滤权重为正的元素，再取前top个并收集为Vec<Cell>：

pub fn select_preference(&mut self) -> Vec<Cell> {
    let top = 3;
    self.sort();
    
    self.cells
        .iter()
        .filter(|cell| cell.weight() > 0)
        .take(top)
        .cloned() // 若Cell未实现Clone，可改用map(|c| c.clone())
        .collect()
}

方案二：分割取片段（适用于0元素集中在末尾）

如果排序后所有权重为0的元素都在末尾，可通过split取第一个非零片段，再截取前top个：

pub fn select_preference(&mut self) -> Vec<Cell> {
    let top = 3;
    self.sort();
    
    self.cells
        .split(|cell| cell.weight() == 0)
        .next() // 获取第一个不含0的切片
        .unwrap_or(&[]) // 处理全为0的边界情况
        .iter()
        .take(top)
        .cloned()
        .collect()
}

补充说明 #

• 若无需转移Cell的所有权，可将返回类型改为Vec<&Cell>，此时去掉.cloned()即可。
• 当前sort方法按权重降序排列，确保了优先级高的元素在前，符合需求。

内容的提问来源于stack exchange，提问作者Mohamad-Jaafar NEHME