如何在Python中实现MATLAB的sortrows(a, [5 -2 4])排序功能

错误原因 #

你现有代码存在两个核心问题：

• np.lexsort的排序键优先级顺序和MATLABsortrows相反，优先级顺序配置错误
• 用[::-1]反转整列的方式无法实现单字段降序逻辑，会打乱行列对应关系，完全不符合排序要求

MATLAB指令逻辑明确 #

sortrows(a, [5 -2 4])的排序规则（优先级从高到低）：

1. 按第5列（MATLAB列索引从1开始）升序排列
2. 第5列值相等时，按第2列降序排列
3. 前两列值都相等时，按第4列升序排列

正确实现代码 #

NumPy适配注意点：

• NumPy列索引从0开始，上述规则对应的列索引为4、1、3
• np.lexsort的键组优先级为从右到左，即最右侧的键是最高优先级排序依据
• 数值列实现降序的正确方式为对列值取负，不要反转整列

import numpy as np

# 键组顺序：优先级低的放前面，优先级高的放后面
# 第2列降序通过取负实现
idx = np.lexsort((a[:,3], -a[:,1], a[:,4]))
a = a[idx]

效果验证 #

我们用小矩阵做跨语言对比验证：

MATLAB测试 #

a = [1 3 5 7 2;
     2 1 4 6 2;
     3 2 3 5 1];
sortrows(a, [5 -2 4])
% 输出：
% 3     2     3     5     1
% 1     3     5     7     2
% 2     1     4     6     2

Python测试 #

a = np.array([[1,3,5,7,2],
              [2,1,4,6,2],
              [3,2,3,5,1]])
idx = np.lexsort((a[:,3], -a[:,1], a[:,4]))
print(a[idx])
# 输出：
# [[3 2 3 5 1]
#  [1 3 5 7 2]
#  [2 1 4 6 2]]

二者输出结果完全一致。

非数值列特殊处理 #

如果待降序的列是字符串等无法取负的非数值类型，可以先对该列做升序排序后反转索引，再结合其他列的排序逻辑处理即可。

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