python - 在for循环内删除np数组中的行

我试图删除列表“ a”中包含的多个2D数组中所有具有一个或更少非零元素的所有行。

当我在“ i”循环外运行该方法时，此方法有效，但不能整体使用。我知道我无法删除要迭代的行，但是我相信在这种情况下我不会这样做，因为我只是删除a中包含的数组中的行，而不是数组本身。

for i in range(len(a)):
  del_idx=[]
  for j in range(len(a[i])):
    nonzero=np.nonzero(a[i][j])
    nonzero_len=len(nonzero[0]) #because np.nonzero outputs a tuple
    if nonzero_len<=1:
        del_idx.append(j)
    else:
        continue
  np.delete(a[i],(del_idx),axis=0)

有人知道这是怎么回事吗？如果这确实不起作用，如何在不使用循环的情况下删除这些元素？这是Python 2.7

谢谢！

最佳答案

当向量化操作可用时，应避免使用NumPy进行for循环。例如，在这里，您可以使用布尔索引：

import numpy as np

np.random.seed(0)

A = np.random.randint(0, 2, (10, 3))

res = A[(A != 0).sum(1) > 1]

array([[0, 1, 1],
       [0, 1, 1],
       [1, 1, 1],
       [1, 1, 0],
       [1, 1, 0],
       [0, 1, 1],
       [1, 1, 0]])

可以将相同的逻辑应用于数组列表中的每个数组。