python - Numpy 数组根据条件返回行

Question

我想返回第二个值大于倒数第二个值的 numpy 数组的行。例如

[[5 0 3 3] 
 [7 9 3 5] 
 [2 4 7 6] 
 [8 8 1 6]]

会回来

[[7 9 3 5]
 [8 8 1 6]]

谢谢！

Answer 1

创建第 2 列大于倒数第 2 列的掩​​码：

mask = a[:,1] > a[:,-2]

# array([False,  True, False,  True])

并用掩码索引：

a[mask]

# array([[7, 9, 3, 5],
#        [8, 8, 1, 6]])

为了清楚起见，这被分成了几个步骤，但当然你可以将它们组合成一个表达式a[a[:,1] > a[:,-2]]。