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

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

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甚至没有必要。

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

出去：

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

Answer 2

您的方法是正确的，它类似于Schwartzian 变换或Decorate-Sort-Undecorate (DSU) 习语

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