如何从Rust ndarray中获取最大值？（无原生max等方法）

和Python的NumPy ndarray不同，Rust的ndarray crate确实没有内置max/min/argmax/argmin这类直接方法，但针对f32/f64类型数组，有几种实用的解决方式：

1. 手动遍历计算（无需额外依赖） #

利用数组的迭代器结合fold方法，手动维护当前最大值：

use ndarray::Array;

pub fn main() {
    let x = Array::from(vec![-3.4, 1.2, 8.9, 2.0]);
    
    // 初始值设为负无穷，确保能覆盖所有数组元素
    let max_val = x.iter().fold(f32::NEG_INFINITY, |current_max, &num| current_max.max(num));
    println!("数组最大值: {}", max_val); // 输出 8.9
}

如果是f64数组，只需将f32::NEG_INFINITY替换为f64::NEG_INFINITY即可。

2. 多维数组的轴上计算 #

如果处理多维数组，可以通过axis_iter或fold_axis按指定维度计算最大值：

use ndarray::{Array, Axis};

pub fn main() {
    // 2x2的f64数组
    let x = Array::from(vec![[-3.4, 1.2], [8.9, 2.0]]).into_shape((2, 2)).unwrap();
    
    // 计算全局最大值
    let global_max = x.iter().fold(f64::NEG_INFINITY, |acc, &val| acc.max(val));
    println!("全局最大值: {}", global_max); // 输出 8.9
    
    // 按行计算最大值
    let row_maxes = x.axis_iter(Axis(0))
        .map(|row| row.iter().fold(f64::NEG_INFINITY, |acc, &val| acc.max(val)))
        .collect::<Vec<_>>();
    println!("每行最大值: {:?}", row_maxes); // 输出 [1.2, 8.9]
}

3. 使用ndarray_stats扩展（最简洁方案） #

官方推荐的ndarray_stats扩展 crate 提供了完整的统计工具，直接支持max、argmax等方法：
首先在Cargo.toml中添加依赖：

[dependencies]
ndarray = "0.15"
ndarray-stats = "0.5"

然后在代码中调用：

use ndarray::Array;
use ndarray_stats::QuantileExt;

pub fn main() {
    let x = Array::from(vec![-3.4, 1.2, 8.9, 2.0]);
    
    // 获取最大值，空数组会返回None，需处理
    let max_val = x.max().unwrap();
    println!("数组最大值: {}", max_val); // 输出 8.9
    
    // 获取最大值的索引（argmax）
    let max_index = x.argmax().unwrap();
    println!("最大值索引: {}", max_index); // 输出 2
}

注意：如果数组为空，max()和argmax()会返回None，实际开发中建议用match或if let做安全处理，避免直接unwrap触发panic。

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