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

Question

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

Answer 1

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]

Answer 2

只是一个 NumPy 替代方案：

import numpy as np

top_2_idx = np.argsort(a)[-2:]
top_2_values = [a[i] for i in top_2_idx]