python - 使用 numpy 用图案掩盖图像？

Question

我正在使用 numpy 来构建像素数组。一个 800x600 的图像是一个 uint8、800x600x3 的 3 维数组。我也有一个具有固定模式的类似数组(棋盘格，参见 here)。我有另一个数组，800x600 的掩码值。在掩码为零的情况下，我想将图案像素复制到图像像素。在掩码不为零的情况下，我想不理会图像像素。

>>> image.shape
(800, 600, 3)
>>> chex.shape
(800, 600, 3)
>>> mask.shape
(800, 600)

感觉应该可以了:

image[mask == 0,...] = chex

但给出“ValueError: array is not broadcastable to correct shape”。

如何将 chex 像素复制到掩码为零的图像像素？

Answer 1

idx=(mask==0)
image[idx]=chex[idx]

请注意，image 的形状为 (800,600,3)，而 idx 的形状为 (800,600)。 The rules用于索引状态

if the selection tuple is smaller than n, then as many : objects as needed are added to the end of the selection tuple so that the modified selection tuple has length N.

因此，索引数组具有自己的广播能力。 idx 的形状被提升为 (800,600,:)