译者:飞龙
协议:CC BY-NC-SA 4.0
到目前为止,我们主要关注使用 NumPy 访问和操作数组数据的工具。本节介绍与 NumPy 数组中的值的排序相关的算法。...然后,如果需要,可以使用这些索引(通过花式索引)构造有序数组:
x[i]
# array([1, 2, 3, 4, 5])
沿行或列的排序
NumPy 排序算法的一个有用特性是,能够使用axis参数来排序多维数组的特定行或列...7, 6, 7],
[1, 2, 4, 5, 7, 7],
[0, 1, 4, 5, 9, 5]])
'''
结果是一个数组,其中每行中的前两个槽包含该行中的最小值,其余值填充剩余的槽...使用转换的成对的平方距离,我们现在可以使用np.argsort对每行排序。...最左边的列将给出最近邻居的索引:
nearest = np.argsort(dist_sq, axis=1)
print(nearest)
'''
[[0 3 9 7 1 4 2 5 6 8]
[1