python - 按2列排序np 2d数组（升序，降序）

Question

我有以下代码/数据：

import numpy as np

data = np.array([
    [12, 1, 0.7, 0],
    [13, 2, 0.5, 1],
    [41, 3, 0.3, 0],
    [20, 2, 0.8, 0],
    [51, 2, 0.7, 0],
    [32, 0, 0.9, 0],
    [17, 4, 0.7, 1],
    [54, 1, 0.6, 1],
    [22, 1, 0.4, 1],
    [74, 0, 0.5, 0],
    [11, 4, 0.3, 1],
    [13, 3, 0.8, 0],
    [15, 4, 0.3, 0],
    [65, 3, 0.2, 1],
])

我想对二维数组进行排序：主要是按 data[:, 1] 升序（从低到高），其次是 data[:, 2] 降序（从高到低） 所以我想出了以下代码：

data[:, 2] = -data[:, 2]
ind = np.lexsort((data[:, 2], data[:, 1]))
data[:, 2] = -data[:, 2]
data = data[ind]
print data

结果：

[[ 32.    0.    0.9   0. ]
 [ 74.    0.    0.5   0. ]
 [ 12.    1.    0.7   0. ]
 [ 54.    1.    0.6   1. ]
 [ 22.    1.    0.4   1. ]
 [ 20.    2.    0.8   0. ]
 [ 51.    2.    0.7   0. ]
 [ 13.    2.    0.5   1. ]
 [ 13.    3.    0.8   0. ]
 [ 41.    3.    0.3   0. ]
 [ 65.    3.    0.2   1. ]
 [ 17.    4.    0.7   1. ]
 [ 11.    4.    0.3   1. ]
 [ 15.    4.    0.3   0. ]]

它是正确的。但我想知道是否有更好的方法来做到这一点。首先，如果有可能在更短的运行时间内完成。第二，更简单的pytonic代码。

为了让它更短（并且更 Pythonic），我可以这样做：

ind = np.lexsort((-data[:, 2], data[:, 1]))
data = data[ind]

运行时仍未得到答复。

Answer 1

您可以直接将取反的第二列与np.lexsort 一起使用-

data[np.lexsort((-data[:, 2], data[:, 1]))]

假设第一列中的非负整数值和第二列中的非负值，这是argsort 的替代方法-

data[(data[:,1]*(data[:,2].max()+1) - data[:,2]).argsort()]

如果第二列总是有[0,1)中的元素，我们可以简单一点-

data[(data[:,1] - data[:,2]).argsort()]