根据行逐步对 Numpy Python 矩阵进行排序

Question

我四处寻找并试图找到一个看似简单问题的解决方案，但一无所获。问题是逐步根据其列对矩阵进行排序。所以，如果我有一个像这样的 numpy 矩阵：

import numpy as np
X=np.matrix([[0,0,1,2],[0,0,1,1],[0,0,0,4],[0,0,0,3],[0,1,2,5]])
print(X)
[[0 0 1 2]
 [0 0 1 1]
 [0 0 0 4]
 [0 0 0 3]
 [0 1 2 5]]

我想根据第一列排序，然后是第二列，第三列，依此类推，得到如下结果：

Xsorted=np.matrix([[0,0,0,3],[0,0,0,4],[0,0,1,1],[0,0,1,2],[0,1,2,5]])
print(Xsorted)
[[0,0,0,3]
 [0,0,0,4]
 [0,0,1,1]
 [0,0,1,2]
 [0,1,2,5]]

虽然我认为可以通过命名列和所有这些来对这样的矩阵进行排序，但我更希望有一种不太依赖于矩阵大小的排序方法。我正在使用 Python 3.4，如果这很重要的话。

任何帮助将不胜感激！

Answer 1

它不会特别快，但你总是可以将你的行转换为元组，然后使用 Python 的排序：

np.matrix(sorted(map(tuple, X.A)))

您也可以使用np.lexsort，如this answer 到somewhat related question 中的建议：

X[np.lexsort(X.T[::-1])]

lexsort 方法似乎更快，但您应该使用实际数据进行测试以确保：

In [20]: X = np.matrix(np.random.randint(10, size=(100,100)))

In [21]: %timeit np.matrix(sorted(map(tuple, X.A)))
100 loops, best of 3: 2.23 ms per loop

In [22]: %timeit X[np.lexsort(X.T[::-1])]
1000 loops, best of 3: 1.22 ms per loop

Answer 2

这里：

data = [[0,0,1,2],[0,0,1,1],[0,0,0,4],[0,0,0,3],[0,1,2,5]]
x  = pandas.DataFrame(data)
# order of columns to sort
z = x.sort([0,1,2,3])
output = z.as_matrix()

output:

array([[0, 0, 0, 3],
   [0, 0, 0, 4],
   [0, 0, 1, 1],
   [0, 0, 1, 2],
   [0, 1, 2, 5]])