Pobierz indeks górnych wartości n listy w pytonie

Question

Mam listę powiedzieć a = [5,3,1,4,10]. Potrzebuję uzyskać indeks dwóch najwyższych wartości z listy viz 5 i 10. Czy istnieje jeden linser, który python oferuje dla takiego przypadku?

Answer 1

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

lub

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

lub

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

lub (w przypadku długich list)

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

Answer 2

Tylko numpy alternatywne:

import numpy as np 

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