Ottieni l'indice dei primi n valori di una lista in python

Question

Ho una lista dire a = [5,3,1,4,10]. Ho bisogno di ottenere un indice dei primi due valori della lista viz 5 e 10. Esiste un one-liner che python offre per un caso del genere?

Answer 1

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

o

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

o

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

o (per le liste lunghe)

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

Answer 2

Solo un NumPy alternative:

import numpy as np 

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