数组元素和

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)，那么我得到（4x2x2x2）+（4x2x2x2）的和。你知道吗

如何对第一个数组中的元素求和，与sum（）函数所做的相反。比如第一组（2-3）=-1，第二组（3+2）=5，等等

谢谢

Answer 1

沿第三个轴求和应满足您的要求：

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