sum() 메서드에 axis 인자를 전달하면, shape에서 그 axis를 제외한 tuple이 결과 array 객체의 shape이 된다.

• (2, 3, 4)에서 axis 0 방향으로 sum() → (3, 4)
• (2, 3, 4)에서 axis 1 방향으로 sum() → (2, 4)

import numpy as np

arr = np.array([i for i in range(24)]).reshape(2, 3, 4)

print("arr")
print(arr)
print("\\narr.sum(0))")
print(arr.sum(0), arr.sum(0).shape)
print("\\narr.sum(1))")
print(arr.sum(1), arr.sum(1).shape)
print("\\narr.sum(2))")
print(arr.sum(2), arr.sum(2).shape)

arr
[[[ 0  1  2  3]
  [ 4  5  6  7]
  [ 8  9 10 11]]

 [[12 13 14 15]
  [16 17 18 19]
  [20 21 22 23]]]

arr.sum(0))
[[12 14 16 18]
 [20 22 24 26]
 [28 30 32 34]] (3, 4)

arr.sum(1))
[[12 15 18 21]
 [48 51 54 57]] (2, 4)

arr.sum(2))
[[ 6 22 38]
 [54 70 86]] (2, 3)