按元素将蒙版划分为其他蒙版

Question

我有一个任意大小的 python 数组掩码，其中包括 (0,1,2,3) 数字。我想把它分成 4 个子掩码，每个子掩码都包含布尔值，第一个在 0 元素位置为真，第二个在 1 元素位置为真，依此类推。我的输入数组太大，我需要一种快速的方法在 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 npmask = [[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