10

有没有办法使用键(或比较器)函数对 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 ]

这是现实的吗?

4

2 回答 2

5

您的方法是正确的,它类似于Schwartzian 变换Decorate-Sort-Undecorate (DSU) 习惯用法

正如我所说,您可以使用 numpy 函数np.argsort。它完成你的工作order_to_index

于 2012-10-13T23:42:49.323 回答
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甚至没有必要。

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

出去:

array([[ 0,  1],
       [ 4, -5],
       [ 2,  3]])
于 2021-04-29T17:07:59.147 回答