Numpy element-wise addition with multiple arrays(多个数组的块元素加法)
问题描述
我想知道是否有更有效的方法来添加多个数字数组(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).
这篇关于多个数组的块元素加法的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!