如何设置ndarray::Array2二维矩阵的单个元素

要修改Array2的元素，首先需要将数组声明为可变绑定（let mut），这是Rust的基本规则——不可变绑定无法修改内容。以下是几种常用的设置元素的方法：

方法1：直接使用二维索引语法 #

这是最直观的方式，Array2支持[[行索引, 列索引]]的索引格式，直接赋值即可：

use ndarray::Array2;

fn main() {
    // 声明为可变绑定才能修改元素
    let mut my_array = Array2::zeros((10, 10));
    
    // 设置第0行第0列的元素为10.0
    my_array[[0, 0]] = 10.0;
    // 设置第3行第5列的元素为3.14
    my_array[[3, 5]] = 3.14;
}

方法2：使用get_mut方法 #

get_mut方法返回Option<&mut T>类型（避免索引越界直接panic），需要先处理Option，再通过解引用修改元素。你创建的Dim对象需要先设置具体的索引值：

use ndarray::{Array2, Dim};

fn main() {
    let mut my_array = Array2::zeros((10, 10));
    
    let mut index = Dim::zeros(2);
    // 将索引设置为(2, 4)
    index[0] = 2;
    index[1] = 4;
    
    // 安全处理Option，避免越界panic
    if let Some(elem) = my_array.get_mut(&index) {
        *elem = 20.0;
    }
    
    // 若确定索引合法，也可以用unwrap直接解包（越界会panic，谨慎使用）
    *my_array.get_mut(&index).unwrap() = 25.0;
}

方法3：通过行/列的可变引用修改 #

如果需要批量操作某一行或列，可以用row_mut/column_mut获取可变行/列引用，再设置元素：

use ndarray::Array2;

fn main() {
    let mut my_array = Array2::zeros((10, 10));
    
    // 获取第5行的可变引用，设置该行第3列的元素为100.0
    my_array.row_mut(5)[3] = 100.0;
    
    // 获取第7列的可变引用，设置该列第2行的元素为50.0
    my_array.column_mut(7)[2] = 50.0;
}

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