仅将项目保留在另一个大小不同的另一个数组中-不循环

时间:2018-07-21 14:43:04

标签: python numpy

让我们说:

e = np.array([1,2,-4])
f = np.array([1,3,1,45,-2,-5,-4,3,3,3,2,-4])

我想获得一个新数组,该数组仅包含f中也出现在e中的元素:

result = np.array([1,1,-4,2,-4])

什么是非循环方式?

谢谢你们。

1 个答案:

答案 0 :(得分:2)

使用np.isin()

>>> f[np.isin(f,e)]
array([ 1,  1, -4,  2, -4])