如何用NumPy排序数组获取最小坐标点？解析轴排序含义

嘿，刚接触NumPy的话，这种需要保持点的完整性的多条件排序确实容易搞晕，我来一步步给你讲清楚！

先搞懂numpy.sort的轴含义 #

你疑惑的轴（axis）其实是指定排序的方向：

• 当axis=0时，NumPy会沿着列的方向排序——简单说就是把每一列单独从小到大排列，完全不管行与行之间的对应关系；
• 当axis=1时，是沿着行的方向排序，把每一行的元素单独排序。

比如你直接对数组用np.sort(arr, axis=0)，得到的结果会是每列各自排好序，但原来的(x,y)点的对应关系就乱了，这显然不是你想要的，因为你需要保持每个点的x和y是一对的。

实现先按x、再按y的升序排序 #

要实现你的需求（保持每个点的完整性，先按x升序，x相同则按y升序），最简洁的方法是用numpy.lexsort，它专门处理多关键字排序：

import numpy as np

# 你的原始数组
arr = np.array([[0.1, 0.2], [0.5, 0.1], [0.3, 0.9], [0.7, 1.0], [0.1, 0.3], [0.2, 0.3]])

# 使用lexsort：参数顺序是【次关键字在前，主关键字在后】
# 这里主关键字是第一列（x），次关键字是第二列（y）
sort_indices = np.lexsort((arr[:, 1], arr[:, 0]))

# 根据索引获取排序后的数组
sorted_arr = arr[sort_indices]

print(sorted_arr)

运行后会得到你想要的结果：

[[0.1 0.2]
 [0.1 0.3]
 [0.2 0.3]
 [0.3 0.9]
 [0.5 0.1]
 [0.7 1.0]]

为什么这么写？ #

np.lexsort((arr[:,1], arr[:,0]))的逻辑是：

1. 首先以arr[:,0]（第一列，x坐标）作为主排序键，按升序排列；
2. 当遇到x值相同的元素时，再用arr[:,1]（第二列，y坐标）作为次排序键，继续按升序排列，完美匹配你的需求。

另一种分步实现方式（适合理解） #

如果你想更直观地理解排序过程，可以用两次稳定排序（稳定排序会保留相同主关键字下元素的原有顺序）：

# 第一步：先按y坐标升序排序（用mergesort保证稳定）
sorted_by_y = arr[arr[:, 1].argsort(kind='mergesort')]
# 第二步：再按x坐标升序排序，此时相同x的元素会保持之前的y顺序
final_sorted = sorted_by_y[sorted_by_y[:, 0].argsort(kind='mergesort')]

这个方法的结果和lexsort完全一致，不过lexsort更高效简洁，适合直接使用。

内容的提问来源于stack exchange，提问作者malhi3