過濾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 ，它會給你你想要的。