获取列表的前N个值的索引

获取列表的前N个值的索引,第1张

获取列表的前N个值的索引
sorted(range(len(a)), key=lambda i: a[i])[-2:]

要么

sorted(range(len(a)), key=lambda i: a[i], reverse=True)[:2]

要么

import operatorzip(*sorted(enumerate(a), key=operator.itemgetter(1)))[0][-2:]

或(对于长列表),请考虑使用

heapq.nlargest

zip(*heapq.nlargest(2, enumerate(a), key=operator.itemgetter(1)))[0]


欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/zaji/5647885.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-12-16
下一篇2022-12-16

发表评论

登录后才能评论

评论列表(0条)

    保存