一旦我有了迭代器向量,如何将其转换为向量迭代器?
创建迭代器有两种方法:使用现有的迭代器适配器或实现自定义迭代器。
让我们采用第二种方法并定义一个采用迭代器向量的自定义迭代器类型:
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]