更新 2D numpy 数组值

Question

有没有更有效的方法来更新多维 numpy 数组的值？
例如，我有一个循环

        for i in np.arange(5):
            for j in np.arange(5):  
                if (i + j) % 2 == 0:   
                    v[i,j] = v[i,j] + v[i, j + 1]  

我正在考虑稍后（使用multiprocessing和Pool）并行化这个过程，但我无法想象如何实现。也许定义一个函数并使用map，但这是一个二维数组，操作取决于元素索引。

Answer 1

基本上你正在这样做：

您可以使用切片索引在两行中完成此操作：

v[0:5:2,0:5:2] += v[0:5:2,1:6:2]    # even rows
v[1:5:2,1:5:2] += v[1:5:2,2:6:2]    # odd rows