python - 在 Python 中对二维数组进行部分 argsort

Question

我有一个 KNN，我需要对列表进行部分argsort。

以下是它现在在代码中的工作原理:

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

kplus_1_nearest_classes = self.trainingY[sorted_distance_indices]

...etc.

我找到了这个答案，How can I partially sort a list?但我不知道如何为 argsort 任务调整“堆化”(而且我不知道如何在 Python 中进行语言互操作，所以我不知道如何手动执行堆排序 alg)...

Answer 1

我想我已经找到答案了。

运行:

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

完成工作。打开以更快的方式。