Entspricht Numpy.argsort() dem grundlegenden Python?

Question

Gibt es eine eingebaute Funktion von Python, die auf python.array tut, was argsort() auf einem numpy.array tut?

Answer 1

ich die oben genannten Vorschläge und meine Ergebnisse hier sind zeitlich fädeln.

Zunächst einmal die Funktionen:

def f(seq): 
    # http://stackoverflow.com/questions/3382352/equivalent-of-numpy-argsort-in-basic-python/3383106#3383106 
    #non-lambda version by Tony Veijalainen 
    return [i for (v, i) in sorted((v, i) for (i, v) in enumerate(seq))] 

def g(seq): 
    # http://stackoverflow.com/questions/3382352/equivalent-of-numpy-argsort-in-basic-python/3383106#3383106 
    #lambda version by Tony Veijalainen 
    return [x for x,y in sorted(enumerate(seq), key = lambda x: x[1])] 


def h(seq): 
    #http://stackoverflow.com/questions/3382352/equivalent-of-numpy-argsort-in-basic-python/3382369#3382369 
    #by unutbu 
    return sorted(range(len(seq)), key=seq.__getitem__) 

nun die IPython Sitzung:

In [16]: seq = rand(10000).tolist() 

In [17]: %timeit f(seq) 
100 loops, best of 3: 10.5 ms per loop 

In [18]: %timeit g(seq) 
100 loops, best of 3: 8.83 ms per loop 

In [19]: %timeit h(seq) 
100 loops, best of 3: 6.44 ms per loop 

FWIW

Answer 2

Es gibt keine integrierte Funktion, aber es ist einfach eine von den tollen Werkzeugen zu montieren Python zur Verfügung stellt:

def argsort(seq): 
    # http://stackoverflow.com/questions/3071415/efficient-method-to-calculate-the-rank-vector-of-a-list-in-python 
    return sorted(range(len(seq)), key=seq.__getitem__) 

x = [5,2,1,10] 

print(argsort(x)) 
# [2, 1, 0, 3] 

Es ist auf Python arbeitet array.array s die gleiche Art und Weise:

import array 
x = array.array('d', [5, 2, 1, 10]) 
print(argsort(x)) 
# [2, 1, 0, 3] 

Answer 3

Meine Alternative mit enumerate:

def argsort(seq): 
    return [x for x,y in sorted(enumerate(seq), key = lambda x: x[1])] 

seq=[5,2,1,10] 
print(argsort(seq)) 
# Output: 
# [2, 1, 0, 3] 

Bessere obwohl Antwort zu verwenden, aus https://stackoverflow.com/users/9990/marcelo-cantos beantworten python sort without lambda expressions

[i for (v, i) in sorted((v, i) for (i, v) in enumerate(seq))] 

Answer 4

diese Frage gefunden, aber benötigt argsort für eine Liste von Objekten basierend auf eine Objekteigenschaft

Erweiterung unutbu Antwort, dies wäre:

sorted(range(len(seq)), key = lambda x: seq[x].sort_property)