Posortuj listę, a następnie podaj indeksy elementów w oryginalnej kolejności.

Question

Mam tablicę liczb n, na przykład [1,4,6,2,3]. Posortowana tablica jest [1,2,3,4,6], a indeksy tych liczb w starej tablicy to 0, 3, 4, 1 i 2. Jaki jest najlepszy sposób, biorąc pod uwagę tablicę n liczb, aby znaleźć tablicę indeksów?

Mój pomysł to uruchomienie statystyk zamówień dla każdego elementu. Jednak ponieważ muszę wielokrotnie przepisywać tę funkcję (w konkursie), zastanawiam się, czy jest na to krótka droga.

Answer 1

>>> a = [1,4,6,2,3] 
>>> [b[0] for b in sorted(enumerate(a),key=lambda i:i[1])] 
[0, 3, 4, 1, 2] 

Objaśnienie:

enumerate(a) zwraca wyliczanie nad krotek składających się z indeksów i wartości w oryginalnej listy: [(0, 1), (1, 4), (2, 6), (3, 2), (4, 3)]

Następnie sorted z key z lambda i:i[1] rodzaju opartych na oryginalnych wartościach (pozycja 1 z każdej krotki).

Wreszcie, zrozumienie listy [b[0] for b in ... ] zwraca oryginalne indeksy (pozycja 0 każdej krotki).

Answer 2

Oto kolejny sposób:

>>> sorted(xrange(len(a)), key=lambda ix: a[ix]) 
[0, 3, 4, 1, 2] 

Takie podejście nie sortuje pierwotną listę, ale jej indeksy (utworzony z xrange), używając oryginalnej listy jako klucze sortowania.

Answer 3

Używanie numpy tablic zamiast list może być korzystne, jeśli wykonujesz wiele statystyk danych. Jeśli zdecydujesz się to zrobić, to będzie działać:

import numpy as np 
a = np.array([1,4,6,2,3]) 
b = np.argsort(a) 

argsort() może działać na listach, jak również, ale wierzę, że w tym przypadku po prostu kopiuje dane do tablicy pierwszy.

Answer 4

To powinno załatwić sprawę:

from operator import itemgetter 
indices = zip(*sorted(enumerate(my_list), key=itemgetter(1)))[0]