gpt4 book ai didi

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

转载 作者:IT老高 更新时间:2023-10-28 20:32:54 27 4
gpt4 key购买 nike

我有一个列表说 a = [5,3,1,4,10]。我需要获取列表前两个值的索引,即 510 我会得到 [0, 4]。 Python 是否为这种情况提供了一种单行代码?

最佳答案

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

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

import operator

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

或者(对于长列表),考虑使用 heapq.nlargest

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

关于python - 获取列表前 N 个值的索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/13070461/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com