使列表分隔值

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