[python] 2차원 배열과 for문

2차원 배열에서 홀수번째 행에서는 우측으로 이동하며 숫자를 적고, 짝수번째 행에 대해서는 좌측으로 이동하며 숫자를 적는 방법

 

1. 2차원 배열 생성 
2. 홀수 행, 짝수 행에 대해 원소 지정
3. 원소 프린트
   
   

---

n = 4
arr_2d = [
    [0 for _ in range(n)]
    for _ in range(n)
]

num = 1
for i in range(n):
    if i % 2 == 0:
        for j in range(n):
            arr_2d[i][j] = num
            num += 1
    else:
        for j in range(n - 1, -1, -1):
            arr_2d[i][j] = num
            num += 1

# 출력
for row in arr_2d:
    for elem in row:
        print(elem, end=" ")
    print()