调用max()后遍历map对象的for循环无输出,求原因与解决办法
Python迭代器耗尽问题分析与解决
问题根源
你代码里的arr = map(int, set(input().split()))生成的是迭代器对象,Python的迭代器有个核心特性:只能被遍历一次。
当调用max(arr)时,会完整遍历迭代器里的所有元素来找出最大值,此时迭代器内部的指针已经走到末尾,后续再用for i in arr遍历,就没有剩余元素可以读取,导致循环无法执行。
解决方案
把迭代器转换成可重复遍历的容器(比如列表、集合)即可,这类容器会把所有元素存储在内存中,支持多次访问:
方案1:转成列表
修改代码如下:
arr = list(map(int, set(input().split()))) print(max(arr)) for i in arr: print(i)
方案2:直接使用集合(简化写法)
原代码里先把输入转成字符串集合再转int有点多余,可直接生成整数集合:
arr = set(map(int, input().split())) print(max(arr)) for i in arr: print(i)
注:集合是无序容器,遍历顺序可能和输入顺序不一致,但不影响功能实现。
补充说明
map()函数返回的是惰性迭代器,它不会一次性生成所有元素,而是在每次迭代时才生成一个元素,遍历完成后就无法再次使用。而列表、集合这类容器会提前把所有元素加载到内存中,因此支持多次遍历、多次调用max()等操作。
内容的提问来源于stack exchange,提问作者Gloryjaw
相关产品推荐
相关产品推荐

