多个数组的块元素加法
问题描述
我想知道是否有更有效的方法来添加多个数字数组(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).
相关文章