多个数组的Numpy元素加法

wnvonmuf  于 2023-03-02  发布在  其他
关注(0)|答案(2)|浏览(115)

我想知道是否有一种更有效的方法来添加多个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个数组。

g2ieeal7

g2ieeal71#

np.sum(list_of_arrays, axis=0)

应该可以或者

np.add.reduce(list_of_arrays).
nkkqxpd9

nkkqxpd92#

最简单的Python解决方案就是使用sum(),如下所示:

sum(list_of_arrays)

相关问题