列出单独的值

Question

我有一个numpy数组：

[[[116 114]
 [111 104]]

有没有办法做到这一点：

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

或使命令总和的命令如下：

[[[230]
 [214]]

如果我使用np.sum()，则返回444。

Answer 1

指定axis：

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

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

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

Answer 2

使用数组tolist函数：

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