使列表分隔值

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