ndarray::Array1中的元素比较生锈了

Question

我试图找到这个 python numpy 代码的 rust 等效项，对数组进行元素比较。

import numpy as np
np.arange(3) > 1

这是我的 Rust 代码：

use ndarray::Array1;
fn main() {
    let arr = Array1::<f64>::range(0.0, 3.0, 1.0);
    arr > 1.0 // doesn't work.
}

我可以使用 for 循环来完成此操作，但我正在寻找最惯用的方法。

Answer 1

使用 .map 对于按元素操作：

fn main() {
    let v: Vec<_> = (0..3).map(|x| x > 1).collect();
    println!("{v:?}")
}

输出：

[false, false, true]

游乐场

Answer 2

答案直接来自map 方法的文档：
arr.map(|x| *x > 1.0)

游乐场