索引numpy的多維數組

Question

我需要僅在最後一列是0的行訪問此numpy的陣列，有時，有時行，其中最後一列的值爲1

y = [0 0 0 0 
    1 2 1 1 
    2 -6 0 1 
    3 4 1 0] 

我有一遍又一遍地做這個，但寧願迴避創建重複數組或每次重新計算。有什麼方法可以確定有關指標並只是給他們打電話？所以我可以這樣做：

>>print y[LAST_COLUMN_IS_0] 
[0 0 0 0 
3 4 1 0] 

>>print y[LAST_COLUMN_IS_1] 
[1 2 1 1 
2 -6 0 1] 

P.S.數組中的列數永遠不會改變，它總是有4列。

Answer 1

您可以使用numpy的boolean indexing來確定要選擇哪些行，並使用numpy的花哨索引/切片來選擇整行。

print y[y[:,-1] == 0, :] 
print y[y[:,-1] == 1, :] 

可以y[:,-1] == 0和... == 1保存像往常一樣，因爲他們只是numpy的陣列。

（該y[:,-1]選擇整個的最後一列，而==相等性檢查發生逐元素，產生布爾值的數組。）