如何用Rust的nalgebra实现类NumPy的逐元素布尔运算与索引？

需要实现和以下NumPy代码等价的逻辑——通过逐元素比较生成布尔条件，并用该条件筛选元素，替换原向量中符合条件位置的元素。目前已用循环实现，但希望找到更简洁优雅的写法。

NumPy 示例代码 #

import numpy as np

value = np.array([1.0, 2.0, 3.0])
error = np.array([0.01, 0.01, 0.01])

current_value = np.array([2.0, 3.0, 4.0])
current_error = np.array([0.1, 0.001, 0.001])
    
improved = current_error < error
value[improved] = current_value[improved]
error[improved] = current_error[improved]

print(error)
print(value)

当前Rust循环实现 #

use nalgebra::{dvector};

fn main() {
    let mut value = dvector![1.0, 2.0, 3.0];
    let mut error = dvector![0.01, 0.01, 0.01];

    let current_value = dvector![2.0, 3.0, 4.0];
    let current_error = dvector![0.1, 0.001, 0.001];
    
    for i in 0..error.len() {
        if current_error[i] < error[i] {
            value[i] = current_value[i];
            error[i] = current_error[i];
        }
    }

    println!("{}", error);
    println!("{}", value);
}

优雅的实现方式 #

利用nalgebra提供的Zip工具（需导入nalgebra::Zip），可以实现元素级的多向量遍历，写法更简洁且符合NumPy的向量式风格：

use nalgebra::{dvector, Zip};

fn main() {
    let mut value = dvector![1.0, 2.0, 3.0];
    let mut error = dvector![0.01, 0.01, 0.01];

    let current_value = dvector![2.0, 3.0, 4.0];
    let current_error = dvector![0.1, 0.001, 0.001];
    
    // 同时遍历四个向量的对应元素
    Zip::from(&mut value)
        .and(&current_value)
        .and(&mut error)
        .and(&current_error)
        .for_each(|val, curr_val, err, curr_err| {
            if curr_err < err {
                *val = curr_val;
                *err = curr_err;
            }
        });

    println!("{}", error);
    println!("{}", value);
}

核心优势 #

• Zip会自动校验所有参与遍历的向量长度一致性，避免手动索引可能引发的越界问题
• 写法贴近NumPy的向量式操作逻辑，无需手动管理循环索引
• 性能与手动循环相当，nalgebra内部对Zip做了针对性优化

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