python - 在 numpy 中使用多个排序键进行排序

Question

我还没有在 SO 上找到这个答案，所以我在这里分享:

问题:当有多个排序键时，如何在 matlab 中模拟 sortrows 功能？在 matlab 中，这看起来像例如:

sortrows(x,[3,-4])

首先按第 3 列排序，然后按第 2 列排序。

如果您按一列排序，您可以使用 np.argsort 找到该列的索引，并应用这些索引。但是如何为多个列执行此操作？

Answer 1

语法很笨拙而且看起来很奇怪，但最干净的做法是 np.lexsort .

data = np.array([[3, 0, 0, .24],
                 [4, 1, 1, .41],
                 [2, 1, 1, .63],
                 [1, 1, 3, .38]]) #imagine rows of a spreadsheet
#now do sortrows(data,[3,-4])
ix = np.lexsort((data[:, 3][::-1], data[:, 2])) 
#this yields [0, 2, 1, 3]

#note that lexsort sorts first from the last row, so sort keys are in reverse order

data[ix]