Python中map对象调用max函数出现异常结果的原因探究

问题场景 #

这段代码意图从输入中找出第二大数，但直接使用map(int, input().split())的返回值时，无法得到预期结果（比如输入1 2 3 4 5时，输出不是4），只有把arr转成list后才能正常运行：

if __name__ == '__main__':
    arr = map(int, input().split())
    a = max([x for x in arr if x != max(arr)])
    print(a)

核心原因：map返回的迭代器只能遍历一次 #

• map()函数返回的是迭代器对象，不是可重复访问的容器。迭代器的本质是“用完即空”——第一次遍历会把所有元素取出来，之后再遍历就没有元素了。
• 原代码里，max(arr)会先完整遍历一遍arr迭代器，找出最大值（比如示例中的5），这时候迭代器已经被耗尽，里面没有剩余元素了。
• 接下来的列表推导式[x for x in arr if x != max(arr)]再去遍历arr，只能得到空列表。对空列表调用max()要么抛出ValueError，要么根本得不到有效结果，自然和预期的4不符。

转成list就正常的原因 #

• 当执行arr = list(arr)时，会把迭代器里的所有元素一次性提取出来，存成一个list。list是可重复遍历的容器，不管遍历多少次，里面的元素都不会消失。
• 这时候max(arr)遍历list找到最大值后，列表推导式再次遍历list，能拿到所有除最大值外的元素，再调用max()就能正确得到第二大数。

直观示例验证 #

输入1 2 3 4 5：

• 用迭代器时：max(arr)遍历完所有元素得到5，迭代器空了；列表推导式得到空列表，max()报错。
• 转成list后：arr是[1,2,3,4,5]，max(arr)得到5，列表推导式筛选出[1,2,3,4]，max()得到4，符合预期。

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