[파이썬] 문제 : nxn 다차원 배열에서 0,1이 체크판 패턴 만들기

import numpy as np

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

data = []
st = 1
for i in range(n):                   # n번 반복
    tmp = [0 for a in range(n)]      # 전부 0으로 채움
    for j in range(len(tmp)):        # len(tmp) = n 번 반복.
        if st == 1:                  # st는 한 row에서 1부터 시작할 것인지, 0부터 시작할 것인지 구분자
            if j%2 == 0:             # 0,2,4와 같이 짝수 인덱스(순서) 일때 1
                tmp[j] = 1
        elif st == 2:                # 1,3,5와 같이 홀수 인덱스(순서) 일때 1
            if j%2 != 0:
                tmp[j] = 1
    if st == 1:                      # 매 row마다 1투버 시작할 것인지 0부터 시작할 것인지 변경
        st = 2
    else:
        st = 1
    data.append(tmp)                 # data 리스트에 row 데이터(리스트) 추가

np_data = np.array(data)             # numpy로 변경
print(np_data)