如何便捷获取Rust中Vec<Vec<T>>二维矩阵的元素？

针对Vec<Vec<T>>类型的二维矩阵，有几种简洁安全的方式获取元素，替代嵌套的get()调用：

1. 链式调用get()或and_then #

直接对行和列连续调用get()，利用Rust的Option链式特性：

let matrix = vec![vec![1, 2], vec![3, 4]];
// 使用?运算符（仅在Result/Option上下文可用）
let elem = matrix.get(0)?.get(1); // 得到Some(&2)

// 通用场景用and_then串联
let elem = matrix.get(0).and_then(|row| row.get(1));

这种方式返回Option<&T>，和一维Vec的get()行为保持一致，避免越界panic。

2. 实现自定义扩展方法 #

如果需要更直观的调用方式，可以给Vec<Vec<T>>实现一个扩展trait：

trait MatrixAccess<T> {
    fn get_2d(&self, row: usize, col: usize) -> Option<&T>;
    fn get_2d_mut(&mut self, row: usize, col: usize) -> Option<&mut T>;
}

impl<T> MatrixAccess<T> for Vec<Vec<T>> {
    fn get_2d(&self, row: usize, col: usize) -> Option<&T> {
        self.get(row).and_then(|row| row.get(col))
    }

    fn get_2d_mut(&mut self, row: usize, col: usize) -> Option<&mut T> {
        self.get_mut(row).and_then(|row| row.get_mut(col))
    }
}

// 使用示例
let mut matrix = vec![vec![1, 2], vec![3, 4]];
let elem = matrix.get_2d(1, 0); // Some(&3)
let elem_mut = matrix.get_2d_mut(0, 1); // Some(&mut 2)

这样可以用get_2d(row, col)直接获取元素，代码可读性更高。

3. 模式匹配处理不存在的情况 #

如果需要分别处理行不存在或列不存在的场景，用嵌套的if let模式匹配：

let matrix = vec![vec![1, 2], vec![3, 4]];
if let Some(row) = matrix.get(2) {
    if let Some(val) = row.get(0) {
        println!("元素存在：{}", val);
    } else {
        println!("列索引越界");
    }
} else {
    println!("行索引越界");
}

注意事项 #

• 避免直接使用matrix[row][col]索引语法，因为索引越界会直接panic，而get()系列方法是安全的，通过Option优雅处理越界情况。
• 如果需要可变引用，对应使用get_mut()链式调用或自定义的get_2d_mut方法。

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