您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

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

5b51 2022/1/14 8:20:27 python 字数 1173 阅读 478 来源 www.jb51.cc/python

假设我有一个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?) 您为每列构建一个布尔数组: co

概述

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]

总结

以上是编程之家为你收集整理的python – 提取Nx3数组的特定行,其中每列满足条件全部内容,希望文章能够帮你解决python – 提取Nx3数组的特定行,其中每列满足条件所遇到的程序开发问题。


如果您也喜欢它,动动您的小指点个赞吧

除非注明,文章均由 laddyq.com 整理发布,欢迎转载。

转载请注明:
链接:http://laddyq.com
来源:laddyq.com
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。


联系我
置顶