numpy iterating over multimimensional array

Question

Я очень новичок в numpy, и я пытаюсь добиться следующего в наиболее pythonic образом. Итак, у меня есть два массива:

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

Теперь, что я хотел бы для каждого элемента Ь, для быть один больше, чем значение соответствующего элемента в т Ь = а + 1

I было интересно, как это можно добиться в 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