如何在Python中屏蔽第二个数组中不存在的数组元素？

Question

这就是我目前拥有的：

x = range(10)
x2 = array(x)
y = [3,6,8]

for i in range(len(x)):
     x2[i] = x2[i] in y

x = ma.masked_where(x2 == False, x)

这让我得到了我想要的东西，但我想这样做而不需要循环。 有没有办法掩盖y中不存在值的数组x？

Answer 1

使用列表推导的这种方法不依赖于任何numpy构造。 它保留了数字，如果它在y ，并且如果不是则放入-

x = [i if i in y else '-' for i in range(10)]

产量

['-', '-', '-', 3, '-', '-', 6, '-', 8, '-']

如果要更改蒙版的默认值，请更改“ - ”。

Answer 2

您可以使用set.intersection获取匹配的数字，它将返回匹配的值，并且无需循环即可完成：

x = range(10)
y = [3,6,8]
s1 = set(x)
s2 = set(y)
s1.intersection(s2)

(Output:) set([8, 3, 6])

x = [1,2,3,4,5]
y = [2,3,6,7,8]
s1 = set(x)
s2 = set(y)
s1.intersection(s2)

(Output:) set([2, 3])

您可以通过转换集来获取列表：

y = list(s1.intersection(s2))

Answer 3

您可以使用numpy为您执行循环，从而加速计算。

x = np.arange(10)
y = np.array([3,6,8])

mask = np.all( x!=y[:,None], 0 )
x = np.ma.masked_where(mask,x)