过滤numpy数组的numpy数组

2024-03-28 09:23:55 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个np数组,我想删除所有符合条件的元素。我希望避免for循环,以使if更快。在

我的np数组的大小是N。在内部,有大小为2的数组。我想删除第一个元素等于零的所有大小为2的数组。在

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

我希望有一个解决方案只需要一个类似的代码。我就是搞不清这条线。我希望有人能帮助我。在

谢谢。在


Tags: 代码元素forifnp数组解决方案remove
1条回答
网友
1楼 · 发布于 2024-03-28 09:23:55

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

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

这里arr[:,0]对应于第一个元素/列或每个子数组。然后检查arr[:,0] !=0,它将返回第一个元素不等于零的那些索引。然后这些索引会自动传递给arr,这会给你想要的。在

相关问题 更多 >