假设我有一个Nx3的数组,我想要满足的元素说：
4 < col1 < 13,5 > col2 > 3,10 > col3 > 6
即应用于：
1,2,3 4,5,6 9,4,7
然后它将提供一个Mx3数组,它只包含满足所有三个条件的行.对于上面的例子,它输出
9,7
我想做一个循环,但我认为numpy或类似的东西必须有更快的东西(例如np.where？)
解决方法
您为每列构建一个布尔数组：
cond1 = (my_array[:,0] > 4) & (my_array[:,0] < 13) cond2 = (my_array[:,1] > 3) & (my_array[:,1] < 5) cond3 = (my_array[:,2] > 6) & (my_array[:,2] < 10)
然后你和你的三个条件在一起：
cond = cond1 & cond2 & cond3
最后你用它来从你的数组中选择行：
out = my_array[cond]