Problem

8 /12


대각선 배열

Theory Click to read/hide

대각선 배열

 diag(V, k=0) 함수를 사용하면 배열에서 대각선을 추출하고 1차원 배열에서 대각선 배열을 만들 수 있습니다.
V - 배열과 같은 개체, 2차원 또는 1차원 배열, 행렬, 목록 또는 튜플 또는 목록이나 튜플을 반환하는 메서드가 있는 모든 함수 또는 개체입니다.
k - 대각선의 인덱스(선택 사항).
기본값은 주 대각선에 해당하는 k = 0입니다. 양수 k 값은 대각선을 위로 이동하고 음수 값은 아래로 이동합니다.

이 함수는 배열 NumPy(ndarray) - 지정된 대각선 배열 또는 지정된 1차원 배열의 대각선 배열을 반환합니다.

Problem

입력은 숫자 n입니다. nxn 크기의 배열을 출력합니다. 대각선 숫자는 0에서 n-1이고 나머지 숫자는 0입니다. .
 

 

예
<헤드> <몸>

time 1000 ms
memory 256 Mb
Rules for program design and list of errors in automatic problem checking

Teacher commentary

Your Program

Attach a file with the source code of the program:

or select a programming language and type your program

Auto

       

Program check result To check the solution of the problem, you need to register or log in!

# 입력 출력
1 5 [[0 0 0 0]
 [0 1 0 0 0]
 [0 0 2 0 0]
 [0 0 0 3 0]
 [0 0 0 4]]