Teilweise Argsort ein 2D-Array in Python

Question

Ich habe eine KNN, und ich brauche teilweise eine Argsort eine Liste.

Hier ist, wie es jetzt in Code funktioniert:

sorted_distance_indices = distances.argsort(axis=1)[:,:self.parameters['k']+1] 

kplus_1_nearest_classes = self.trainingY[sorted_distance_indices] 

...etc. 

ich diese Antwort gefunden, How can I partially sort a list? Aber ich sehe nicht, wie die ‚heapification‘ für die argsort Aufgabe anzupassen, (und ich habe keine Idee, wie man Spracheninterop in Python tut, also sehe ich nicht, wie man den heapsort alg manuell tut) ...

Answer 1

Ich denke, dass ich die Antwort habe.

Running:

sorted_distance_indices = np.argpartition(distances,self.parameters['k']+1,axis=1)[:,:self.parameters['k']+1] 

Ruft den Job zu erledigen. Offen für einen schnelleren Weg.