译者:飞龙
协议:CC BY-NC-SA 4.0
到目前为止,我们主要关注使用 NumPy 访问和操作数组数据的工具。本节介绍与 NumPy 数组中的值的排序相关的算法。...np.partition接受一个数组和一个数字K;结果是一个新数组,最小的K个值在分区左边,任意顺序的剩下的值在右边:
x = np.array([7, 2, 3, 1, 6, 5, 4])
np.partition...(x, 3)
# array([2, 1, 3, 4, 6, 5, 7])
请注意,结果数组中的前三个值是数组中的三个最小值,其余数组位置包含其余值。...示例:K 最近邻
让我们快速了解如何沿着多个轴使用这个argsort函数,来查找集合中每个点的最近邻居。我们首先在二维平面上创建一组 10 个随机点。...回想一下,两点之间的平方距离是每个维度的平方差的总和;使用由 NumPy 提供的,高效广播(“数组计算:广播”)和聚合(“聚合:最小值,最大值和之间的一切”)的例程,我们可以在一行代码中计算平方距离矩阵