NumPy:按np.array过滤行

rkjt50r983

我想通过检查另一个数组是否包含列值来过滤NumPy 2-d数组。我怎样才能做到这一点?

import numpy as np

ar = np.array([[1,2],[3,-5],[6,-15],[10,7]])
another_ar = np.array([1,6])
new_ar = ar[ar[:,0] in another_ar]
print new_ar

我希望得到的[[1,2],[6,-15]]只是上面的代码[1,2]

卡斯拉文

您可以使用np.where,但是请注意,ar[:,0]如果ar需要循环遍历并检查成员资格,则它是第一元素的列表

>>> ar[np.where([i in another_ar for i in ar[:,0]])]
array([[  1,   2],
       [  6, -15]])

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章