[Python] 각 행 또는 열의 합 구하기

[[1 2 3]
 [4 5 6]
 [7 8 9]]

일때, 

 

a = arr.sum(axis=0)

print(a)

b = arr.sum(axis=1)

print(b)

 

결과는 아래와 같다.

a
[12 15 18]

b
[ 6 15 24]