【发布时间】:2021-05-12 15:38:19
【问题描述】:
我想知道是否有更高效/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 matrix numpy-ndarray numpy-ufunc