Rust ndarray使用axis_iter_mut修改数组时类型不匹配问题求助

你遇到的核心问题是可变视图与所有权数组的类型不兼容：

• axis_iter_mut(Axis(1))生成的迭代器元素是ArrayBase<ViewRepr<&mut i32>, _>——这是原数组c的列的可变引用视图，仅借用原数组内存，无独立所有权。
• 而d是ArrayBase<OwnedRepr<i32>, _>——拥有独立内存的所有权数组，Rust不允许直接将所有权类型赋值给引用视图类型。

要实现用d更新c的每一列，需使用ndarray提供的元素复制方法，而非直接赋值：

修改后的代码 #

use ndarray::{array, Axis, ArrayBase, Dim, OwnedRepr};

fn main() {
    let mut c: ArrayBase<OwnedRepr<i32>, Dim<[usize; 2]>> = array![
        [1, 2],
        [1, 2]
    ];
    let d: ArrayBase<OwnedRepr<i32>, Dim<[usize; 1]>> = array![1, 1];

    // 使用assign方法将d的元素复制到每一列的可变视图中
    c.axis_iter_mut(Axis(1)).for_each(|mut x| x.assign(&d));

    println!("{:?}", c);
    // 输出: [[1, 1], [1, 1]]
}

关键说明 #

• assign(&d)方法会遍历d的所有元素，逐个复制到当前可变视图x（即c的某一列）的对应位置，完全适配视图的类型要求。
• 若偏好底层操作，也可使用copy_from_slice，需先将d转为切片：x.copy_from_slice(d.as_slice().unwrap());，效果与assign一致。

这种方式完全符合你要求的axis_iter_mut遍历更新逻辑，可无缝扩展到批量矩阵乘法场景——后续只需在for_each闭包中替换为矩阵乘法逻辑，将结果赋值给对应视图即可。

内容的提问来源于stack exchange，提问作者ginger314