python - 为 numpy 数组的每个 "other"字段 ((i+j)%2==0) 添加值

Question

我有一个 m-by-n numpy 数组，我想将 1.0 添加到所有条目 [i, j] 当 (i + j) % 2 == 0 时，即“每隔一个方格”。

我当然可以简单地遍历字段

import numpy as np

a = np.random.rand(5, 4)

for i in range(a.shape[0]):
    for j in range(a.shape[1]):
        if (i + j) % 2 == 0:
            a[i, j] += 1.0

但不用说这真的很慢。

知道如何改进吗？

Answer 1

你可以很容易地分两步完成操作，比如

import numpy as np

a = np.zeros((5, 14))

# Even rows, odd columns
a[::2, 1::2] += 1
# Odd rows, even columns
a[1::2, ::2] += 1

print a