Numpy：按键排序功能答案

【问题标题】：Numpy: sort by key functionNumpy：按键排序功能
【发布时间】：2012-10-04 08:56:44
【问题描述】：

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

这是现实的吗？

【问题讨论】：

order_to_index?你的意思是argsort？
@wiso 确实如此！我从来没有想过这与给出对键值向量排序并按它排序的索引是同一个问题！
试试看：np.argsort([3,6,2,2,8]) #-> array([2, 3, 0, 1, 4])。它不会对您的原始输入进行排序。通常 numpy 函数没有副作用
@wiso 确实，刚刚做了，我在此期间编辑了我的最后一篇文章，抱歉。谢谢！您会考虑添加答案吗？

【解决方案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]])

【讨论】：

【解决方案2】：

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

【讨论】：