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