numpy數組的列表索引如何工作？

Question

我有一小段代碼，我想了解如何使用另一個numpy數組作爲另一個numpy數組的索引。

bmi = np.array([13,14,15,22]) 
light = np.array(bmi) < 21 
print(bmi[light]) 

輸出：[13,14,15]

我瞭解，numpy的陣列工作時，事情完成逐元素，所以我的體重指數[光]假設是，它只會打印bmi中的對應值具有光的真值。

Answer 1

如果您想打印light，你會得到：

[ True True True False] 

這是一個布爾面具。

現在當您使用print(bmi[light])時，您將bmi過濾爲True。

您可能已經反轉，使用~口罩，只得到[22]：

print(bmi[~light]) 

相同評論中指出，你可以瞭解更多關於布爾屏蔽here。