假设我有一个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?)
答案 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]