список делают отдельные значения

Question

У меня есть Numpy массив:

[[[116 114] 
[111 104]] 

Есть ли способ сделать это:

[[[116,114] 
    [111,104]] 

или команду, чтобы сделать сумму этого выглядеть следующим образом:

[[[230] 
[214]] 

Если я использую np.sum(), он возвращает 444.

Answer 1

Использование массива tolist функции:

>>> r = numpy.array([[116, 114],[111, 104]]) 
>>> [sum(e) for e in r.tolist()] 
[230, 215] 

Answer 2

Укажите axis:

>>> a = np.array([[[116,114], [111,104]]]) 

>>> np.sum(a, axis=-1) 
array([[230, 215]]) 

>>> a.sum(axis=-1) 
array([[230, 215]])