중첩 루프와 루프 패턴

소개

기본 중첩 반복문

for i in range(3):
    for j in range(3):
        print(i, j)

구구단 표

for i in range(1, 4):
    for j in range(1, 4):
        print(f'{i*j:3}', end='')
    print()

삼각형 패턴

for i in range(1, 6):
    print('*' * i)

행렬 순회

matrix = [[1,2,3],[4,5,6]]
for row in matrix:
    for val in row:
        print(val, end=' ')
    print()

누적 패턴

matrix = [[1,2],[3,4]]
total = sum(v for row in matrix for v in row)
print(total)

요소 찾기

matrix = [[1,2,3],[4,5,6],[7,8,9]]
for i, row in enumerate(matrix):
    for j, val in enumerate(row):
        if val == 5:
            print(f'Found at ({i},{j})')

중복 없는 조합

items = ['a','b','c']
for i in range(len(items)):
    for j in range(i+1, len(items)):
        print(items[i], items[j])

중첩 반복문의 효율성

# Efficient: use a set for inner lookup
data = {1, 2, 3, 4}
result = [x for x in range(10) if x in data]  # O(n)
print(result)

중첩 컴프리헨션으로 평탄화하기

matrix = [[1,2],[3,4]]
flat = [x for row in matrix for x in row]
print(flat)

중첩 반복문에서 빠져나오기

def find_in_matrix(matrix, target):
    for i, row in enumerate(matrix):
        for j, val in enumerate(row):
            if val == target:
                return (i, j)
    return None
print(find_in_matrix([[1,2],[3,4]], 3))

빠른 확인

복습

계속 진행하기