在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)。