[파이썬] 문제: numpy로 nxn 배열 만들

사용자로부터 2이상의 수 n을 입력으로 받아서, 
입력된 수를 바탕으로 다음과 같은 nxn 크기의 다차원 배열 a를 생성하는 
프로그램을 작성하시오, 
이때 배열의 내용은 0과 1의 값이 체크 판 패턴으로 교차하여 나타나도록 해라.


실행결과
n을 입력하시오 : 4
[[1 0 1 0]
[0 1 0 1]
[1 0 1 0]
[0 1 0 1]]

실행결과
n을 입력하시오 : 5
[[1 0 1 0 1]
[0 1 0 1 0]
[1 0 1 0 1]
[0 1 0 1 0]
[1 0 1 0 1]]

 

import numpy as np

def create_checkerboard(n):
    matrix = np.zeros((n, n), dtype=int)          # nxn 차월의 배열을 만들고 0으로 채움

    matrix[1::2, ::2] = 1                          # 1로 채우기
    matrix[::2, 1::2] = 1                          # 1로 채우기

    return matrix

n = int(input("n을 입력하시오: "))    # 입력 받기
result = create_checkerboard(n)       # 배열 생성 함수 실행
print(result)