COS Pro 2급 파이썬

이제 반복문을 사용하여 2차원 리스트의 요소를 모두 출력해보겠습니다. 먼저 for 반복문을 한 번만 사용하는 방식입니다.

>>> a = [[10, 20], [30, 40], [50, 60]]
>>> for x, y in a:    # 리스트의 가로 한 줄(안쪽 리스트)에서 값 두 개를 꺼냄
...     print(x, y)
...
10 20
30 40
50 60

2차원 리스트에 for를 사용하면 가로 한 줄씩 반복하게 됩니다. 전체 리스트를 기준으로 보면 안쪽 리스트가 통째로 반복됩니다. 이 때 for x, y in a:와 같이 in 앞에 변수를 두 개 지정해주면 가로 한 줄(안쪽 리스트)에서 값 두 개를 꺼내 옵니다.

당연히 in 앞에 변수의 개수는 2차원 리스트에서 가로 크기(안쪽 리스트의 요소 개수)와 일치해야겠죠? for 반복문을 한 번만 사용하는 방식은 2차원 리스트의 가로 크기가 크지 않을 때 유용합니다.

이번에는 for 반복문을 두 번 사용해서 2차원 리스트의 요소를 출력해보겠습니다. 다음 내용을 IDLE의 소스 코드 편집 창에 입력한 뒤 실행해보세요.

two_dimensional_list_for_for.py

a = [[10, 20], [30, 40], [50, 60]]
 
for i in a:        # a에서 안쪽 리스트를 꺼냄
    for j in i:    # 안쪽 리스트에서 값을 하나씩 꺼냄
        print(j, end=' ')
    print()

실행 결과

10 20
30 40
50 60

먼저 for i in a:는 전체 리스트에서 가로 한 줄씩 꺼내 옵니다(안쪽 리스트를 통째로 꺼내 옴). 다시 for j in i:와 같이 가로 한 줄(안쪽 리스트) i에서 값을 하나씩 꺼내면 됩니다.

그럼 이번에는 for range에 세로 크기와 가로 크기를 지정해서 2차원 리스트의 요소를 인덱스로 접근해보겠습니다.

two_dimensional_list_for_for_range.py

a = [[10, 20], [30, 40], [50, 60]]
 
for i in range(len(a)):            # 세로 크기
    for j in range(len(a[i])):     # 가로 크기
        print(a[i][j], end=' ')
    print()

실행 결과

10 20 
30 40 
50 60

for range에 세로 크기와 가로 크기를 넣으면 인덱스로 사용할 수 있습니다. 여기서 주의할 점은 len으로 2차원 리스트 a의 크기를 구하면 리스트 안에 들어있는 모든 요소의 개수가 아니라 안쪽 리스트의 개수(세로 크기)가 나온다는 점입니다. 즉, len(a)는 6이 아니라 3입니다. 그리고 len으로 안쪽 리스트 a[i]의 크기를 구해야 안쪽 리스트에 들어있는 요소의 개수(가로 크기)가 나옵니다. 즉, len(a[i])는 2입니다.

따라서 for i in range(len(a)):는 세로 크기 3만큼 반복하고, for j in range(len(a[i])):는 가로 크기 2만큼 반복합니다. 요소에 접근할 때는 a[i][j]처럼 리스트[세로인덱스][가로인덱스] 형식으로 접근합니다.

이번에는 while 반복문을 사용하여 2차원 리스트의 요소를 출력해보겠습니다.

two_dimensional_list_while.py

a = [[10, 20], [30, 40], [50, 60]]
 
i = 0
while i < len(a):    # 반복할 때 리스트의 크기 활용(세로 크기)
    x, y = a[i]
    print(x, y)
    i += 1           # 인덱스를 1 증가시킴

실행 결과

10 20
30 40
50 60

while 반복문을 사용할 때도 리스트의 크기를 활용하면 편리합니다. 여기서도 len(a)처럼 2차원 리스트의 크기를 구했으니 안쪽 리스트의 개수(세로 크기)가 나오겠죠? 따라서 6이 아닌 3이 나옵니다.

리스트에 인덱스를 지정하여 값을 꺼내 오는데 다음과 같이 변수 두 개를 지정해주면 가로 한 줄(안쪽 리스트)에서 값 두 개를 한꺼번에 가져올 수 있습니다.

x, y = a[i]

값을 가져왔으면 반드시 i += 1과 같이 인덱스를 1 증가시켜 줍니다.

그럼 while 반복문을 두 번 사용할 수도 있겠죠?

two_dimensional_list_while_while.py

a = [[10, 20], [30, 40], [50, 60]]
 
i = 0
while i < len(a):           # 가로 크기
    j = 0
    while j < len(a[i]):    # 세로 크기
        print(a[i][j], end=' ')
        j += 1              # 가로 인덱스를 1 증가시킴
    print()
    i += 1                  # 세로 인덱스를 1 증가시킴

실행 결과

10 20 
30 40 
50 60

먼저 while i < len(a):와 같이 세로 크기만큼 반복하면서 while j < len(a[i]):와 같이 가로 크기(안쪽 리스트의 요소 개수)만큼 반복하면 됩니다. 가로 크기는 len(a[i])와 같이 안쪽 리스트의 크기를 구하면 되겠죠?

요소에 접근할 때는 리스트[세로인덱스][가로인덱스] 형식으로 접근하면 됩니다. 따라서 세로 인덱스로 변수 i를, 가로 인덱스로 변수 j를 지정합니다.

print(a[i][j], end=' ')

값을 가져왔으면 반드시 j += 1과 같이 가로 인덱스를 1 증가시켜 줍니다. 마찬가지로 바깥쪽 while에서는 i += 1과 같이 세로 인덱스를 1 증가시켜 줍니다. 이때 i += 1을 안쪽 while에서 수행하면 반복이 제대로 되지 않으므로 주의해야 합니다.

i = 0
while i < len(a):
    j = 0
    while j < len(a[i]):
        print(a[i][j], end=' ')
        j += 1
        i += 1    # 안쪽 while에서 i를 증가시키면 안 됨. 잘못된 방법
    print()