python - 将 numpy 数组垂直添加到自身

Question

我正在尝试垂直向自身添加一个 (174, 2, 2) 数组。到目前为止，我能够以非 numpy 的方式做到这一点:

import numpy as np
cm1 = np.random.rand(174, 2, 2)
temp_array = np.zeros([2, 2])

for _content in cm1:
    temp_array = np.add(temp_array, _content)

print(temp_array)

我能够得到结果，但是有没有一种简单的方法可以做到这一点？

Answer 1

您可以使用sum参数 axis=0 为:

cm1.sum(axis=0)
array([[82.39817762, 84.41947252],
       [91.82740901, 83.53547764]])