python - 如何将 PyArray 转换为 Vec> 在锈

Question

我正在使用 Rust 中的 numpy crate 来处理来自 python 的二维数组。PyArray（https://docs.rs/numpy/0.11.0/numpy/array/struct.PyArray.html）实现了一个from_vec2()函数，将 aVec<Vec<T>>转换为PyArray<T, 2>（2D PyArray），以及一个to_vec()函数，将 2D 数组扁平化为一维向量并返回Vec<T>，但它没有实现to_vec2()。有没有一种简单的方法来进行我缺少的这种转换，或者我必须手动实现这个功能？

谢谢。

Answer 1

您可以.iter与一些迭代器方法一起使用来执行此操作：

arr.iter().unwrap().map(|arr| arr.to_vec().unwrap()).collect::<Vec<_>>()