numpy - 我如何获得或创建与rust的ndarray crate 等效的numpy的amax函数？

Question

我想使用numpy crate 在Rust中获得或创建amax的ndarray功能。因此，基本上，我希望能够在Rust中获得以下行为:

>>> import numpy as np
>>> arr = np.array(range(1, 28)).reshape((3, 3, 3))
>>> np.amax(arr, axis=1)
array([[ 7,  8,  9],
       [16, 17, 18],
       [25, 26, 27]])

据我所知， amax中没有 ndarray函数。我如何使用 ndarray数组获得与以上示例中提到的功能相同的功能:

let arr = Array3::from_shape_vec((3, 3, 3), (1..28).collect()).unwrap();

Answer 1

您可以使用map_axis和min完成此操作

let a_min = arr.map_axis(Axis(0), |view| *view.iter().min().unwrap() )

您可以在 playground中看到一个有效的示例