[파이썬] 문제 : numpy 2차원 배열의 대각선으로 값 입력

import numpy as np

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

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