numpy删除以零开头的所有行

Question

我有2D numpy数组，我想删除以某些值开头的所有行，然后将所有以其他值开头的行保持在

a1 = np.array([[ 0,  1,  2,  3,  4],
   [ 5,  6,  0,  8,  0],
   [10, 11, 12, 13, 14],
   [ 0, 16, 17, 18, 19],
   [20, 21, 22,  0, 24]])

第一步之后的

a2 = ([[ 5,  6,  0,  8,  0],
      [10, 11, 12, 13, 14],
      [20, 21, 22,  0, 24]])

新数组中，然后将所有行开始。

a3 = ([[10, 11, 12, 13, 14]])

Answer 1

您可以使用以下蒙版来实现此目的：

mask = (a1[:, 0] != 0)
a2 = a1[mask, :]

mask2 = (a2[:, 0] == 10)
a3 = a2[mask2, :]

Answer 2

您可以使用np.logical_not方法来创建要删除的行的条件。

'''

a2 = a1[np.logical_not(a1[:, 0] == 0)]
a3 = a2[np.logical_not(a2[:, 0] != 10)]

'''