在滑动窗口中应用 np.where

Question

我有一个 True/False 值数组，我想将其用作另一个不同形状数组的重复掩码。

import numpy as np

mask = np.array([[ True, True],
                 [False, True]])

array = np.random.randint(10, size=(64, 64))

我想在滑动窗口中应用这个掩码，类似于数组上的where 函数。目前，我使用np.kron 来简单地重复掩码以匹配数组的尺寸：

layout = np.ones((array.shape[0]//mask.shape[0], array.shape[1]//mask.shape[1]), dtype=bool)
mask = np.kron(layout, mask)
result = np.where(mask, array, 255) # example usage

是否有任何优雅的方式来执行相同的操作，而无需将mask 重复为与array 相同的形状？我希望会有某种滑动窗口技术或卷积/相关。

Answer 1

你可以试试np.tile:

np.where(np.tile(mask, (a//m for a,m in zip(array.shape, mask.shape))), array, 255)

Answer 2

使用带有 reshape 的广播，这样您就不需要额外的内存来存储重复的mask：

x, y = array.shape[0]// mask.shape[0], array.shape[1] // mask.shape[1]

result1 = np.where(mask[None, :, None], 
                  array.reshape(x, mask.shape[0], y, mask.shape[1]), 
                  255).reshape(array.shape)