Question

我有一个任意大小的python数组掩码，其中包括（0,1,2,3）个数字。我想将其分为4个子掩码，每个子掩码都包含布尔值，第一个在0元素位置具有True，第二个在1元素位置具有True，依此类推。我的输入数组太大，我需要一种快速的方法来在python中做到这一点。这是一个示例：

mask = [[0 1 1]
        [2 0 3]
        [3,3,2]
        [1,1,1]]

我想要拥有：

sub_mask[0] = [[1 0 0]
               [0 1 0]
               [0 0 0]
               [0 0 0]]
sub_mask[1] = [[0 1 1]
               [0 0 0]
               [0 0 0]
               [1 1 1]]
sub_mask[2] = [[0 0 0]
               [1 0 0]
               [0 0 1]
               [0 0 0]]
sub_mask[3] = [[0 0 0]
               [0 0 1]
               [1 1 0]
               [0 0 0]]

Answer 1

您可以通过简单的列表理解来做到这一点：

sub_mask = [
    mask == val
    for val in np.unique(mask)
]

Answer 2

您可以使用numpy：

import numpy as np

mask = [[0, 1, 1],
        [2, 0, 3],
        [3, 3, 2],
        [1, 1, 1]]
mask = np.array(mask)

def get_sub_mask(n):
    # Create an empty sub-mask with all 0's
    sub_mask = np.zeros(mask.shape)
    # Set elements to 1 based on their position in the original mask
    sub_mask[np.where(mask == n)] = 1
    return sub_mask

将遮罩按其元素划分为其他遮罩

2 个答案: