Python删除numpy数组中的行

Question

我有一个大的 numpy 数组(8 x 30000)，我想根据某些条件删除一些行。此标准仅适用于一列。

例子:

>>> p = np.array([[0, 1, 3], [1 , 5, 6], [4, 3, 56], [1, 34, 4]])
>>> p
array([[ 0,  1,  3],
   [ 1,  5,  6],
   [ 4,  3, 56],
   [ 1, 34,  4]])

在这里，我想删除第 3 列的值大于 30 的每一行，即。这里是第 3 行。

由于数组非常大，我想避免 for 循环。我想到了这个:

>>> a[~(a>30).any(1), :]
array([[0, 1, 3],
   [1, 5, 6]])

但是，它显然删除了最后两行。关于如何以有效的方式做到这一点有什么想法吗？

Answer 1

p = p[~(p[:,2] > 30)]

或(如果您的情况很容易逆转):

p = p[p[:,2] <= 30]

返回

array([[ 0,  1,  3],
       [ 1,  5,  6],
       [ 1, 34,  4]])