通过其他数组给出的索引屏蔽数组

Question

我有以下数组：

a = [10, 31, 30, 11, 17, 12, 22, 25, 85, 17, 21, 43]
b = [0, 1, 4, 6]

我想a根据 array 给出的索引进行掩码b。这意味着得到：

c = [True, True, False, False, True, False, True, False, False, False, False, False]

Answer 1

In [524]: a=np.array([10, 31, 30, 11, 17, 12, 22, 25, 85, 17, 21, 43])
In [525]: b=np.array([0, 1, 4, 6])

要使索引处c的布尔值为 True b，只需使用：

In [526]: c=np.zeros(a.shape, bool)
In [527]: c[b]=True

In [528]: c
Out[528]: 
array([ True,  True, False, False,  True, False,  True, False, False,
       False, False, False], dtype=bool)

然后您可以选择以下值a：

In [529]: a[c]
Out[529]: array([10, 31, 17, 22])

但你也可以选择它们b：

In [530]: a[b]
Out[530]: array([10, 31, 17, 22])

但c更适合删除那些，a[~c]. np.delete(a,b)做同样的事情。

其他生成数组的方法c有

np.in1d(np.arange(a.shape[0]),b)
np.any(np.arange(a.shape[0])==b[:,None],0)

由于我只是在另一个问题中讨论屏蔽数组，所以我可以在这里做同样的事情：

In [542]: np.ma.MaskedArray(a,c)
Out[542]: 
masked_array(data = [-- -- 30 11 -- 12 -- 25 85 17 21 43],
             mask = [ True  True False False  True False  True False False False False False],
       fill_value = 999999)