admin管理员组文章数量:1558092
Python新手在这里,我已经阅读了Filter rows of a numpy array?和doc但仍然无法弄清楚如何以python方式编写代码.
我有的示例数组:(实际数据是50000 x 10)
a = numpy.asarray([[2,'a'],[3,'b'],[4,'c'],[5,'d']])
filter = ['a','c']
我需要在过滤器中找到a中带有[:,1]的所有行.预期结果:
[[2,'a'],[4,'c']]
我目前的代码是这样的:
numpy.asarray([x for x in a if x[1] in filter ])
它工作正常,但我在某处读到它效率不高.什么是适当的numpy方法?
编辑:
谢谢你所有正确的答案!不幸的是,我只能将一个标记为已接受的答案我很惊讶numpy.in1d没有出现在谷歌搜索numpy过滤器2d数组.
解决方法:
您可以使用可以使用np.in1d生成的bool索引数组.
您可以沿着您想要使用的任何axis索引np.ndarray,例如指示是否应包含元素的bool数组.由于你想沿轴= 0索引,这意味着你想从outest索引中选择,你需要有1D np.array,其长度是行数.其每个元素都将指示是否应包含该行.
一个快速的方法是在a的第二列使用np.in1d.您可以通过[:,1]获得该列的所有元素.现在你有一个np.array,其元素应该根据你的过滤器进行检查.这就是np.in1d的用途.
所以完整的代码看起来像:
import numpy as np
a = np.a
版权声明:本文标题:python二维数组筛选_python numpy过滤二维数组的条件 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://m.elefans.com/xitong/1727342027a1109532.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论