python - 在 Numpy 中合并两个相同形状数组的最简洁方法

Question

我们有一些形状相同的数组，想要合并它们。

通过“合并”，我的意思是输出一个新数组，其中每个位置的每个数组中的每个 i,j 的总和。

import numpy as np
first = np.array([[1,1],[1,1]])
second = np.array([[2,2],[2,2]])
third = np.array([[3,3],[3,3]])

结果应该是:

[[6,6],
[6,6]]

这是我的代码...但是有更简洁的方法吗？我似乎找不到内置方法:

def merge_arrays(arrays):
    output = arrays[0]
    for a in arrays[1:]:
        for i,row in enumerate(a):
            for j,col in enumerate(row):
                output[i,j] += a[i,j]
    return output

merge_arrays([first, second, third])

Answer 1

它只是 output = first + second + third 或 np.sum([first, second, third], axis=0)。