Rust中实现类似Python np.zeros的嵌套向量填充函数

首先要解决原代码里的核心问题：Rust是静态类型语言，不能靠运行时判断是否为Vec，得通过类型系统区分嵌套结构和叶子元素；同时要修正可变迭代、泛型约束的问题，以下是可行实现：

trait VectorHelper {
    // 约束填充值可转换为目标类型且能复制，保证每个元素都能拿到独立的值
    fn fill_with(&mut self, value: impl Into<Self> + Clone);
}

// 为所有叶子类型实现：直接赋值
impl<T: Clone> VectorHelper for T {
    fn fill_with(&mut self, value: impl Into<Self> + Clone) {
        *self = value.into();
    }
}

// 为嵌套Vec实现：递归处理每个子元素
impl<T: VectorHelper> VectorHelper for Vec<T> {
    fn fill_with(&mut self, value: impl Into<T> + Clone) {
        // 用iter_mut()获取可变引用，才能修改内部元素
        for elem in self.iter_mut() {
            elem.fill_with(value.clone());
        }
    }
}

代码说明 #

• 泛型约束：impl Into<Self> + Clone 既允许传入同类型值，也支持能转换为目标类型的值，同时Clone保证每个元素都能拿到独立的填充值。
• 叶子类型处理：对所有实现Clone的基础类型（i32、f64、String等），直接完成赋值操作。
• 嵌套结构处理：遍历Vec的可变元素，递归调用fill_with，自动穿透所有嵌套层级直到叶子节点。

使用示例 #

fn main() {
    // 构造一个3层嵌套的Vec
    let mut nested_vec: Vec<Vec<Vec<i32>>> = vec![vec![vec![1,2], vec![3,4]], vec![vec![5,6]]];
    
    // 填充所有元素为0
    nested_vec.fill_with(0);
    
    println!("{:?}", nested_vec);
    // 输出：[[[0, 0], [0, 0]], [[0, 0]]]
}

简化版（仅支持同类型填充） #

如果不需要跨类型转换，可以进一步简化约束：

trait VectorHelper {
    fn fill_with(&mut self, value: impl Clone);
}

impl<T: Clone> VectorHelper for T {
    fn fill_with(&mut self, value: impl Clone) {
        *self = value.clone();
    }
}

impl<T: VectorHelper + Clone> VectorHelper for Vec<T> {
    fn fill_with(&mut self, value: impl Clone) {
        for elem in self.iter_mut() {
            elem.fill_with(value.clone());
        }
    }
}

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