Python3内置max函数处理含NaN数组时输出异常问题

原因分析 #

你的代码返回索引0而非2的核心问题出在NaN的比较特性上：在Python中，NaN（非数值）与任何值进行比较（包括自身）都会返回False。

当max()函数通过key参数比较元素时，它会从第一个元素开始，依次判断后续元素的key值是否大于当前记录的最大值的key值：

• 第一个元素的key是NaN，后续元素的key（比如索引2的3.0）与NaN比较时，3.0 > NaN返回False，因此max()不会更新当前最大值的索引。
• 最终，max()会保留初始的第一个元素索引0。

解决方案 #

要正确找到最大有效元素的索引，需要在key函数中对NaN做特殊处理，将其视为比所有有效数值更小的值（比如负无穷）：

方案1：使用math模块判断NaN #

import math

array = [float('nan'), 1.0, 3.0, 2.0, float('nan')]
max_index = max(range(len(array)), key=lambda i: array[i] if not math.isnan(array[i]) else -float('inf'))
print(max_index)  # 输出2

方案2：利用NaN自身特性判断（无需导入模块） #

array = [float('nan'), 1.0, 3.0, 2.0, float('nan')]
max_index = max(range(len(array)), key=lambda i: array[i] if array[i] == array[i] else -float('inf'))
print(max_index)  # 输出2

这两种方式都会把NaN对应的key设为负无穷，确保有效数值的key始终大于NaN的key，从而让max()正确找到最大有效元素的索引。

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