[파이썬] 문제 : numpy 2차원 배열에서 원소의 전체합,행열 방향 합 구하기

import numpy as np

n = int(input('n을 입력하시오 : '))     # 입력 받기

arr = np.zeros(n*n, dtype='int32') # 전부 0으로 채원진 배열 만들기
arr = arr.reshape(n,n)             # 2차원 배열로 변환
for i in range(len(arr)):         # 반복
    for j in range(i+1):          # i+1 만큼 반복(1로 표시할 인덱스)
        arr[i][j] = 1              # arr[i][j]로 1로 표시할 인덱스 찾고 해당 값을 1 부여
print(arr)

print(np.sum(arr))

print(np.sum(arr, axis=0))

print(np.sum(arr, axis=1))