Numpy: argsort funktioniert nicht für mehrdimensionale Arrays

argsort() funktioniert einwandfrei, aber die Indizierung funktioniert nicht so, wie Sie es erwarten.

http://docs.scipy.org/doc/numpy/reference/arrays.indexing.html#advanced -indexing

Versuche dies:

a = np.random.random([5,5])
i = np.arange(len(a))[:, np.newaxis]
j = np.argsort(a, axis=1)
a_sorted = a[i, j]
np.sort(a, axis=1)

Ich war genauso verwirrt. In der Dokumentation heißt es ausdrücklich,

Kehrt zurück
index_array : ndarray, int
Array von Indizes, die a entlang der angegebenen Achse sortieren.
Mit anderen Worten, a[index_array] ergibt ein sortiertes a .

Das gilt nur für ein 1D-Array. Sollte die Dokumentation dahingehend erweitert werden, dass dies bei ND-Arrays nicht der Fall ist?

@lzkelley Ja, ein Patch zur Verbesserung der Dokumentation wäre sehr willkommen.

@shoyer cool, ich werde eine PR dafür machen und aktualisieren

Es wäre gut, wenn es ein Beispiel dafür gäbe, wie die Ausgabe tatsächlich zum Sortieren des ND-Arrays verwendet wird.

Es wäre auch gut, wenn es eine lesbarere Lösung als a[np.arange(np.shape(a)[0])[:,np.newaxis], np.argsort(a)] gäbe

Vielleicht könnte @seberg es zu einer seiner Indexierungsfunktionsergänzungen machen.

Ich glaube nicht, dass es sehr offensichtlich zur Indizierung passt, eine Auswahlfunktion oder so ist wahrscheinlich eine einfachere Übereinstimmung.

+1 für eine neue Funktion.

Soll nicht der gesamte Zweck von argsort (im Gegensatz zu sort ) für die Indizierung verwendet werden?

argsort liefert die Indizes zum Sortieren entlang einer Achse, aber das Sortieren entlang dieser Achse mithilfe der Indizes erscheint unnötig ausführlich. Es sei denn, es gibt einen einfacheren Weg, der mir nicht bekannt ist.

https://github.com/numpy/numpy/blob/5becd12914d0402967205579d6f59a98151e0d98/doc/neps/indexing.rst#related -questions

eine Auswahlfunktion oder so ist wahrscheinlich eine einfachere Übereinstimmung.

Das ist #8708