如何按内部元素优先级排序二维NumPy数组？无需结构化数组

你可以使用np.lexsort函数实现非结构化数组的多优先级排序，这是NumPy专门针对多键排序场景优化的方法，无需转换为结构化数组。

实现代码 #

import numpy as np

arr = np.array([
    [1, 2, 3],
    [0, 2, 1],
    [1, 1, 2],
    [3, 2, 1],
    [1, 1, 0]
])

# 按第0列（第一个元素）→第2列（第三个元素）→第1列（第二个元素）的优先级排序
# lexsort参数为排序键元组，优先级从右到左：最右侧是主排序键
sorted_indices = np.lexsort((arr[:, 1], arr[:, 2], arr[:, 0]))
sorted_arr = arr[sorted_indices]

print(sorted_arr)

输出结果 #

[[0 2 1]
 [1 1 0]
 [1 1 2]
 [1 2 3]
 [3 2 1]]

原理说明 #

np.lexsort的核心特性是按元组从右到左的顺序确定排序优先级：

1. 最右侧的arr[:, 0]是最高优先级排序键，先按每行第一个元素升序排列
2. 当第一个元素相同时，用倒数第二个键arr[:, 2]（每行第三个元素）继续排序
3. 前两个元素都相同时，用最左侧的arr[:, 1]（每行第二个元素）完成最终排序

你之前的错误分析 #

1. 普通数组使用order参数报错：
   order参数仅适用于结构化数组，它需要指定字段名而非列索引，普通二维数组没有字段结构，因此触发ValueError。

2. 结构化数组尝试失败：
   你设置了axis=1，这会对每行内部的元素排序，而非对整个数组的行进行排序，完全偏离了需求；同时直接将普通二维数组转换为结构化数组的方式也存在数据布局不匹配的问题。

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