多个数组的块元素加法

2022-05-26 00:00:00 numpy matrix numpy-ndarray numpy-ufunc

问题描述

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


解决方案

np.sum(list_of_arrays, axis=0) 

应该行得通。或

np.add.reduce(list_of_arrays). 

相关文章