multidimensional-array - ndarray : Is there an effective way to swap 2 row/cols/etc.在不同的数组？

Question

结合使用带有2个Array结构的ndarray，是否存在一种有效的方法来交换2行/列/(沿某个轴的切片)？

标题确实总结了这个问题。

Answer 1

虽然不同的人通过“有效”可能意味着不同的事情，但是一种直接的方法是通过临时方式将行/列彼此分配。这是两行(Playground)的示例:

let mut a = arr2(&[[1, 2, 3], [4, 5, 6]]);
let mut b = arr2(&[[7, 8, 9], [10, 11, 12], [13, 14, 15]]);
let mut a_row = a.slice_mut(s![1, ..]);
let mut b_row = b.slice_mut(s![2, ..]);
let tmp = a_row.to_owned();
a_row.assign(&b_row);
b_row.assign(&tmp);
println!("a = {:?}", a);
println!("b = {:?}", b);

另外，您可以使用 azip!() 逐元素交换并避免创建临时数组，甚至使用 par_azip! 并行化复制。