如何通过索引条件正确屏蔽 numpy 二维数组？

Question

我有一个 2D numpy 数组。

x = np.arange(25).reshape(5, 5)

我需要按索引条件创建掩码。例如具有偶数索引的元素。

mask = np.zeros((5, 5), dtype=bool)

for i in range(5):
    for j in range(5):
        if i % 2 == 0 and j % 2 == 0:
            mask[i][j] = True

有没有办法使用 numpy 工具创建这样的掩码？

Answer 1

用这个替换你的嵌套循环：

mask[::2,::2] = True 

::2 表示每个轴上的“从头到尾，每隔一个元素”。