获取列表中的最大的N项和最小的N项
获取列表中的最大的N项和最小的N项
#!/sur/bin/env python # -*- coding:utf-8 -*- # author:zengsf #time:2018/10/31 import heapq nums = [1, 0, 2, 29, 7, -7, 18, 23, 5, 20, 9, 10, 12] # 最大的4个数的索引 max_num_index_list = map(nums.index, heapq.nlargest(4, nums)) # 最小的4个数的索引 min_num_index_list = map(nums.index, heapq.nsmallest(4, nums)) print(max_num_index_list) print(list(max_num_index_list)) print(list(min_num_index_list))
输出结果:
<map object at 0x0000021066E627F0> [3, 7, 9, 6] [5, 1, 0, 2]
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步