我有一个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:
如果
arr
是您的输入数组,请执行以下操作:这里
arr[:,0]
对应于第一个元素/列或每个子数组。然后检查arr[:,0] !=0
,它将返回第一个元素不等于零的那些索引。然后这些索引会自动传递给arr
,这会给你想要的。在相关问题 更多 >
编程相关推荐