过滤numpy数组的numpy数组

Question

我有一个np数组数组，我想删除所有匹配条件的元素。我希望避免使用for循环，以使其速度更快。

np数组的形状是N,2。我想移除第一个元素等于零的所有子数组。

[[1,2],
 [0,5], # <--- Remove
 [5,1],
 [0,3], # <--- Remove
 [1,1],
 [0,0], # <--- Remove
 [5,0],
 .....
 [5,5]] 

我希望有一个只有一行代码的解决方案。我就是搞不懂这条线。我希望有人能帮我。

谢谢。

Answer 1

如果arr是您的输入数组，请执行以下操作：

arr = arr[arr[:,0] !=0]

这里，arr[:,0]对应于第一个元素/列或每个子数组。然后检查arr[:,0] !=0，它返回第一个元素不等于零的索引。然后，这些索引将自动传递给arr，这将为您提供所需的信息。