3

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

通过“合并”，我的意思是输出一个新数组，其中每个位置的每个数组中的每个 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])
4

1 回答 1

4

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

于 2013-06-12T15:10:16.787 回答