獲取python列表中前n個值的索引

Question

我有一個列表說a = [5,3,1,4,10]。我需要得到列表5和10中前兩個值的索引。是否有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] 

或

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

或（對於長列表）

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

Answer 2

僅有numpy的替代：

import numpy as np 

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