具有多个数组的 Numpy 元素加法

Question

我想知道是否有更高效/pythonic 的方式来添加多个 numpy 数组 (2D) 而不是:

def sum_multiple_arrays(list_of_arrays):
   a = np.zeros(shape=list_of_arrays[0].shape) #initialize array of 0s
   for array in list_of_arrays:
      a += array
   return a 

Ps:我知道 np.add() 但它只适用于 2 个数组。

Answer 1

np.sum(list_of_arrays, axis=0) 

应该可以。或者

np.add.reduce(list_of_arrays).