python – 提取Nx3数组的特定行,其中每列满足条件

发布时间：2020-12-16 21:37:55 所属栏目：Python 来源：网络整理

导读：假设我有一个Nx3的数组,我想要满足的元素说： 4 col1 13,5 col2 3,10 col3 6 即应用于： 1,2,34,5,69,4,7 然后它将提供一个Mx3数组,它只包含满足所有三个条件的行.对于上面的例子,它输出 9,7 我想做一个循环,但我认为numpy或类似的东西必须有更快的东西(例如

假设我有一个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]

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!