Вертикально добавить массив Numpy к себе

Question

Я пытаюсь добавить массив (174, 2, 2) к себе по вертикали. Пока я могу делать это не просто:

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]])