将子数组加在一起

Question

我有一个包含 40 个元素的 numpy 数组，每个元素有 500 个元素。

我想像这样对每个子数组求和。

 fourier[0] + fourier[1] + fourier[2] + ...

例如，

 fourier[0] = [0,1,2,3,4,5,...]
 fourier[1] = [2,3,4,5,6,8,...]
 fourier_sum = fourier[0] + fourier[1]

因此，fourier_sum =[2,4,6,8,10,13,...]

但是，像上面那样对每个子数组求和是没有意义的。

还有别的办法吗？

谢谢

Answer 1

您可以简单地使用以下使用矢量化运算对每一列求和以生成一维数组系列，然后对系列求和以得出总数：

fourier_sum = np.sum(np.sum(fourier))

基本示例：

fourier = np.array([[1, 2, 3, 4], [4, 5, 6, 7], [8, 9, 10, 11]])

给出 70 的 fourier_sum