我有一个二维数组Array2
,我希望能够 Select 行
use ndarray::{Array2, Axis};
let subarray: Array2 = my_array.select(Axis(0), &[0,2,5]);
有了这段简单的代码,我可以 Select 第一行、第三行和第六行,但是当在运行时构建目标索引时呢?我的意思是,我想从Vector
或任何可以作为函数输入的迭代数中进行 Select .
我有一个二维数组Array2
,我希望能够 Select 行
use ndarray::{Array2, Axis};
let subarray: Array2 = my_array.select(Axis(0), &[0,2,5]);
有了这段简单的代码,我可以 Select 第一行、第三行和第六行,但是当在运行时构建目标索引时呢?我的意思是,我想从Vector
或任何可以作为函数输入的迭代数中进行 Select .
您可以通过简单地borrow 一个向量来获得切片
let index = vec![0, 2, 4];
let subarray = my_array.select(Axis(0), &index);