提取Nx3阵列的特定行,从而每列满足条件

时间:2013-06-20 19:30:30

标签: python arrays numpy

假设我有一个Nx3的数组,我想要满足这样的元素:

4 < col1 < 13, 5 > col2 >  3, 10 > col3 > 6

即。将此应用于:

1,2,3
4,5,6
9,4,7

然后它将提供一个Mx3数组,它只包含满足所有三个条件的行。对于上面的例子,它输出

9,4,7

我想过做一个循环,但我认为numpy或类似的东西必须有更快的东西(例如np.where?)

1 个答案:

答案 0 :(得分:3)

为每列构建一个布尔数组:

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)

然后and你的三个条件:

cond = cond1 & cond2 & cond3

最后用它来选择数组中的行:

out = my_array[cond]