選擇基於第一個

Question

第二陣列比方說，我有len(1000)兩個數組每個

array_a = np.array([1,2,3,....,1000]) # length of 1000 
array_b = np.array([32344,83242,94323,....,48984]) # length of 1000 

現在我選擇基於某些條件的array_a一個子集：

subset_a = array_a[(array_a>10) * (array_a<500)] 

現在如何選擇屬於上述subset_a的array_b的值？

我試圖

subset_b = array_b[subset_a] 

，但我得到一個錯誤

IndexError: arrays used as indices must be of integer (or boolean) type 

Answer 1

你在找這個？

import numpy as np 

array_a = np.array([1,2,3,4,5]) # length of 5 
array_b = np.array([6,7,8,9,10]) # length of 5 

condition = array_a>3 
print condition 
subset_a = array_a[condition] 
print subset_a 
subset_b = array_b[condition] 
print subset_b 

http://ideone.com/dAFLYL