python - 使列表单独的值

Question

我有一个 numpy 数组:

[[[116 114]
 [111 104]]

有没有办法做到这一点:

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

或者一个命令使这个总和看起来像这样:

[[[230]
 [214]]

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

Answer 1

指定轴:

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

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

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