遍历矩阵并连续求和而无需for循环

时间:2019-11-07 10:45:17

标签: python loops numpy matrix sum

我有400个2x2 numpy矩阵,我需要将它们加在一起。我想知道是否有比使用for循环更好的方法,因为它要花费大量时间和内存来遍历for循环,特别是如果我有更多矩阵(将来可能是这种情况)?

1 个答案:

答案 0 :(得分:0)

只是想通了。我所有的矩阵都在列表中,所以我用

np.sum(<list>, axis=0)

它给出了所有400个矩阵之和的2x2矩阵!