numpy迭代多维数组

Question

我对numpy很新，我试图以最pythonic的方式实现以下目标。所以，我有两个数组：

a=array([[0, 1, 2],[3,4,5]])
b=zeros(a.shape)

现在，我希望b中的每个元素都比a中的相应元素的值大一个，即b = a + 1

我想知道如何在numpy中实现这一目标。

Answer 1

最简单的方法如下：

b = a + 1

但是如果你想自己迭代数组（虽然不推荐）：

for i in range(len(a)):
    for j in range(len(a[i])):
        b[i][j] = a[i][j] + 1