[英]Mask for an array in the same shape
Array2[:,0] 包含 array1 行索引,array2[:,1] 包含 array1 元素值。 我想以矢量化的方式獲得與 array1 相同形狀的掩碼。
array1=
[[0 1 2]
[3 4 5]
[6 7 8]]
array2=
[[0 1]
[1 3]
[1 5]
[2 7]
[2 9]]
代碼:
array1 = np.arange(9).reshape(-1,3)
array2 = np.arange(10).reshape(-1,2)
array2[:,0]=[0,1,1,2,2]
print(array1[array2[:, 0]] == array2[:, 1,None])
我得到的結果:
[[False True False]
[ True False False]
[False False True]
[False True False]
[False False False]]
我想得到的結果:
[[False True False]
[ True False True]
[False True False]
編輯:循環解決方案如下所示:
mask=np.zeros_like(array1)
for (y,x) in array2:
mask[y,(np.where(array1[y,:] == x))] = True
您可以執行回映射:
array1 = np.arange(9).reshape(-1,3)
array2 = np.arange(10).reshape(-1,2)
array2[:,0] = [0,1,1,2,2]
xs, ys = np.where(array1[array2[:, 0]] == array2[:, 1,None])
mask = np.zeros_like(array1, dtype=bool)
mask[array2[xs,0], ys] = True
這為我們提供了給定的樣本數據:
>>> mask
array([[False, True, False],
[ True, False, True],
[False, True, False]])
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.