我想通过检查另一个数组是否包含列值来过滤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] 删除。
我来说两句