Как суммировать все массивы в списке массивов?

Я работаю с матрицей путаницы. Поэтому для каждого цикла у меня есть массив (матрица путаницы). Поскольку я делаю 10 циклов, я получаю 10 массивов. Я хочу суммировать их все. Поэтому я решил, что для каждого цикла я буду хранить массивы в списке - я не знаю, лучше ли хранить их внутри массива.

А теперь я хочу добавить каждый массив, который находится внутри списка.

Так что если у меня есть:

    5 0 0       1 1 0
    0 5 0       2 4 0
    0 0 5       2 0 5

Сумма будет:

    6 1 0
    2 9 0 
    2 0 10

Это картина моих матриц путаницы и моего списка массивов:

Это мой код:

   list_cm.sum(axis=0)

1 ответ

Решение

Просто sum список:

>>> sum([np.array([[5,0,0],[0,5,0],[0,0,5]]), np.array([[1,1,0],[2,4,0],[2,0,5]])])
array([[ 6,  1,  0],
       [ 2,  9,  0],
       [ 2,  0, 10]])
Другие вопросы по тегам