在NumPy中,可以使用布爾索引來根據條件篩選數組元素。首先,我們需要創建一個條件數組,然后將這個條件數組作為索引傳遞給原始數組。例如,假設我們有一個一維數組arr,我們想要篩選出其中大于5的元素,可以按照以下步驟操作:
import numpy as np
arr = np.array([1, 6, 4, 9, 3, 8])
# 創建條件數組
condition = arr > 5
# 使用條件數組篩選元素
filtered_arr = arr[condition]
print(filtered_arr)
這將輸出 [6 9 8]
,即原始數組中大于5的元素。您也可以將條件直接傳遞給索引:
filtered_arr = arr[arr > 5]
這樣也可以得到相同的結果。您可以根據需要更改條件,以滿足不同的篩選需求。