파이썬에서리스트의 상위 n 개의 값 인덱스를 얻는다.

Question

나는 목록이 a = [5,3,1,4,10]이라고 말한다. 목록 5와 10의 상위 2 개 값의 인덱스를 가져와야합니다. 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:]]