numpy.extract：從數組中提取符合條件的元素

從數組中提取符合條件的元素

arr = np.arange(10)
arr
#array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
#設置條件，如：能整除3
condition = np.mod(arr, 3)==0
condition
#array([ True, False, False,  True, False, False,  True, False, False,True])
#np.extract()會把True對應的元素提取出來返回
np.extract(condition, arr)
#輸出：array([0, 3, 6, 9])

參考：
[1] Extract：從數組中提取符合條件的元素