在Numpy中合併兩個相同形狀數組的最乾淨的方法

Question

我們有一些相同形狀的數組，並且想合併它們。

通過「合併」，我的意思是輸出一個新的數組，每個位置的每個數組中的每個i，j的總和。

import numpy as np 
first = np.array([[1,1],[1,1]]) 
second = np.array([[2,2],[2,2]]) 
third = np.array([[3,3],[3,3]]) 

結果應該是：

[[6,6], 
[6,6]] 

這裏是我的代碼......但有一個更清潔的方式？我似乎無法找到一個內置的方法：

def merge_arrays(arrays): 
    output = arrays[0] 
    for a in arrays[1:]: 
     for i,row in enumerate(a): 
      for j,col in enumerate(row): 
       output[i,j] += a[i,j] 
    return output 

merge_arrays([first, second, third]) 

Answer 1

這只是output = first + second + third或np.sum([first, second, third], axis=0)。