数组的求和元素

Question

我有一个像这样的数组：

array = np.array([[[[ 2, -3],[ 3,  2]],[[-4, -1],[-5,  1]],
                [[-7, -5],[-1,  6]],[[-5,  0],[-4,  2]]],
               [[[-1,  4],[ 6,  1]],[[-2, -3],[-5,  5]],
                [[-2, -8],[-1,  7]],[[-1,  8],[-4,  2]]]])

如果我sum(array)，则得到的总和为（4x2x2）+（4x2x2）。

我如何求和第一个数组内的元素，与sum（）函数的作用相反。像第一组中的（2-3）= -1，第二组中的（3 + 2）= 5，等等。

谢谢

Answer 1

沿第3轴求和应该可以完成您想要的操作：

res = np.sum(array, axis=3)
# or:
# res = array.sum(axis=3)

产生

[[[ -1   5]
  [ -5  -4]
  [-12   5]
  [ -5  -2]]

 [[  3   7]
  [ -5   0]
  [-10   6]
  [  7  -2]]]