如何在Rust中实现Vec<Vec<u64>>矩阵的列只读迭代器？

需求说明 #

我们需要为Vec<Vec<u64>>类型的矩阵实现一个只读迭代器，效果和Python中list(zip(*m))一致——每次迭代返回矩阵的一列元素，直到所有列遍历完成。由于Rust允许同一数据的多个不可变引用，这个需求完全可以实现。

核心思路 #

通过为矩阵的每一行维护一个不可变迭代器，每次从这些行迭代器中同步取出当前位置的元素引用，组成当前列的元素集合。当任意一行的迭代器耗尽时，整个列迭代器终止（和Pythonzip的行为一致，以最短行为准）。

代码实现 #

定义列迭代器结构体

use std::slice::Iter;

// 列迭代器，持有矩阵每一行的不可变迭代器
struct ColumnIter<'a> {
    row_iters: Vec<Iter<'a, u64>>,
}

impl<'a> ColumnIter<'a> {
    // 从矩阵创建列迭代器
    fn new(matrix: &'a Vec<Vec<u64>>) -> Self {
        // 为每一行生成对应的不可变迭代器
        let row_iters = matrix.iter().map(|row| row.iter()).collect();
        ColumnIter { row_iters }
    }
}

实现Iterator trait

impl<'a> Iterator for ColumnIter<'a> {
    // 迭代器的返回类型：当前列所有元素的不可变引用集合
    type Item = Vec<&'a u64>;

    fn next(&mut self) -> Option<Self::Item> {
        let mut column = Vec::with_capacity(self.row_iters.len());
        
        // 遍历所有行的迭代器，尝试取出下一个元素
        for iter in &mut self.row_iters {
            match iter.next() {
                Some(val) => column.push(val),
                // 只要有一行没有元素，说明所有列已遍历完成
                None => return None,
            }
        }
        
        Some(column)
    }
}

使用示例 #

fn main() {
    let matrix = vec![vec![1, 2], vec![3, 4]];
    let mut col_iter = ColumnIter::new(&matrix);

    // 输出第一列：Some([1, 3])
    println!("{:?}", col_iter.next());
    // 输出第二列：Some([2, 4])
    println!("{:?}", col_iter.next());
    // 输出None，迭代结束
    println!("{:?}", col_iter.next());
}

关键细节说明 #

1. 生命周期管理：'a生命周期确保迭代器持有的元素引用和原矩阵的生命周期绑定，避免出现悬垂引用，符合Rust的安全规则。
2. 行长度不一致的处理：如果矩阵各行长度不同，迭代器会以最短的行为准，和Pythonzip的逻辑完全一致。
3. 灵活调整返回类型：如果需要返回元素值而非引用，可以将Item改为Vec<u64>，并在column.push(*val)处复制元素（对于u64这类小类型，性能开销可以忽略）。
4. 第三方库替代方案：如果允许依赖第三方库，可以使用itertools crate中的multiunzip方法快速实现类似功能，但上述代码是纯标准库实现，无需额外依赖。

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