选择基于第一个

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