Numpy：按键功能排序

Question

有没有办法使用键（或比较器）函数对numpy ndarray的行进行排序，而无需转换为python列表？

特别是，我需要根据这个功能进行排序：

c1,c2= 4,7
lambda row: c1*(row[1]/c2)+row[0]

我意识到一种可能的解决方案是生成一个具有每行键值的向量，但是如何根据它进行排序？是否应该试图以某种方式将这种向量转换为索引向量？

order= c1*(matrix[:,1]/c2)+matrix[:,0]
indexes= order_to_index( order )
return matrix[ indexes ]

这是现实吗？

Answer 1

您的方法是正确的，它类似于Schwartzian transform或Decorate-Sort-Undecorate (DSU) idiom

正如我所说，你可以使用numpy函数np.argsort。它完成了order_to_index。

的工作

Answer 2

为了更明确的答案，假设我们有一个数组 x 并且想要根据某个函数 func 对行进行排序，该函数接受一行 x 并输出一个标量。< /p>

x[np.apply_along_axis(func, axis=1, arr=x).argsort()]

对于这个例子

c1, c2 = 4, 7
x = np.array([
    [0, 1],
    [2, 3],
    [4, -5]
])
x[np.apply_along_axis(lambda row: c1 * / c2 * row[1] + row[0], 1, x).argsort()]

出：

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

在这种情况下，甚至不需要 np.apply_along_axis。

x[(c1 / c2 * x[:,1] + x[:,0]).argsort()]

出：

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