Question

我正在写一个使用Vec<Vec<T>>类型的库，以按列大的顺序存储数据（每个内部Vec代表一列）。用户可以创建具有任何行和列长度的Vec<Vec<T>>，但是所有列都必须具有相同的长度。

有时我需要逐行有效地遍历Vec<Vec<T>>。我不想更改数组类型，因为在大多数情况下，我需要迭代“按列向量”（一次迭代一个完整的列向量）。

除非我丢失了某些内容，否则Iterator::zip是不可行的，因为我事先不知道列向量的数量。 Itertools::izip和Itertools::multizip也不可行。

这是我的示例代码：

let array = vec![vec![1, 2, 3], vec![4, 5, 6], vec![7, 8, 9]];
let mut iterators: Vec<_> = array.iter().map(|x| x.iter()).collect();
for _ in 0..array[0].len() {
    let values: Vec<_> = iterators.iter_mut().map(|x| x.next().unwrap()).collect();
    dbg!(values);
}

我应该在开始迭代之前定义一个可变的values向量，以避免在每个循环中进行分配，还是编译器还是会进行这种优化？自己找到最简单的方法是什么？

有没有更有效/惯用的解决方案？

Answer 1

一旦有了向量的迭代器，如何将其转换为向量的迭代器？

创建迭代器有两种方法：使用现有的迭代器适配器或实现自定义迭代器。

让我们采用第二种方法，并定义一个采用迭代器向量的自定义Iterator类型：

struct DynamicZip<I>
where I: Iterator {
    iterators: Vec<I>
}

，让我们提供一个Iterator实现：

impl<I, T> Iterator for DynamicZip<I>
where I: Iterator<Item = T> {
    type Item = Vec<T>;
    fn next(&mut self) -> Option<Self::Item> {
        let output: Option<Vec<T>> = self.iterators.iter_mut().map(|iter| iter.next()).collect()
        output
    }
}

我们完成了！

回到原始示例

fn main() {
    let array = vec![vec![1, 2, 3], vec![4, 5, 6], vec![7, 8, 9]];
    let iterators: Vec<_> = array.into_iter().map(|v| v.into_iter()).collect();
    let dz = DynamicZip { iterators: iterators };
    // use the Iterator type we just defined
    for column in dz {
        println!("{:?}", column)
    }
}

将产生输出

[1, 4, 7]
[2, 5, 8]
[3, 6, 9]

Answer 2

您的Vec<Vec<T>>是面向列的矩阵，其中每个内部Vec都是一列，因此，可以通过简单的方式在运行时了解的列数执行array.len()操作。

对于行和列，创建Iterator会更容易。这是一个示例：

fn main() {
    let matrix = vec![vec![1, 2, 3], vec![4, 5, 6], vec![7, 8, 9]];

    let columns = matrix.len();
    let rows = matrix[0].len();
    // If you know the number of rows in advance. E.g.: In some constructor
    // let rows = 3;

    let iter = (0..rows).map(|row_idx| matrix.iter().flatten().skip(row_idx).step_by(columns));

    for (row_idx, row_values) in iter.enumerate() {
        for (column_idx, value) in row_values.enumerate() {
            println!("[{}, {}] = {}", row_idx, column_idx, value);
        }
    }
}

如何有效地逐行遍历`Vec <vec <t >>`？

2 个答案:

一旦有了向量的迭代器，如何将其转换为向量的迭代器？

回到原始示例