Python获取列表最大或最小N个元素索引的正确与错误方式
本文介绍在Python中查找列表最大或最小几个数索引的方法。通过对比heapq.nlargest和手动遍历两种实现,指出当列表存在重复元素时,heapq.nlargest结合nums.index会错误地返回相同数字的第一次出现索引,而手动方法能正确区分不同位置的元素。
具体实现方式如下所示:
nums = [1,8,2,23,7,-4,18,23,24,37,2] result = map(nums.index, heapq.nlargest(3, nums)) temp=[] Inf = 0 for i in range(3): temp.append(nums.index(max(nums))) nums[nums.index(max(nums))]=Inf result.sort() temp.sort() print(result) print(temp)
上述代码展示了result和temp两种不同的求解逻辑,运行后的输出结果为:
[3, 8, 9] [3, 8, 9]
结果看起来没有问题。
然而,如果将nums列表修改为包含重复元素的情况:
nums = [1,8,2,23,7,-4,18,23,23,37,2]
此时的输出变为:
[3, 3, 9] [3, 7, 9]
这里暴露出一个问题:使用result方法时,如果遇到相同的数字,它始终只会返回该数字第一次出现的索引位置。
以上就是本文关于Python查找列表最大或最小几个数索引方法的完整分享,希望上述内容能对大家的学习或工作提供有益参考,同时也欢迎大家继续支持本站。
本站声明
本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。













