我想知道是否有更高效/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 个数组。
最佳答案
np.sum(list_of_arrays, axis=0)
应该可以。或者
np.add.reduce(list_of_arrays).
关于具有多个数组的 Numpy 元素加法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66111665/