[파이썬] 문제 : numpy 2차원 배열에서 대각선 방향(X) 값 설정

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)):         # 반복
    arr[i][i] = 1                  # ↘ 방향. arr[i][i]로 대각선 인덱스 찾고 해당 값을 1 부여
for i in range(len(arr)):         # 반복
    arr[i][len(arr)-1-i] = 1      # ↙방향. arr[i][i]로 대각선 인덱스 찾고 해당 값을 부여
print(arr)