파이썬 리스트 컴프리헨션은 어떤 이터러블에서든 읽기 쉬운 한 줄로 새 리스트를 만듭니다. 빈 리스트를 만들고 반복하며 append를 호출하는 대신, 원하는 결과를 서술합니다.

이 가이드는 기본 형태에서 시작해 필터, 중첩 반복문, 그리고 저희가 직접 측정한 시간까지 다룹니다. 파이썬이 처음이라면 초보자를 위한 파이썬 튜토리얼(영문)부터 보시고, 가상 환경 가이드(영문)로 환경을 깔끔하게 유지하세요.

파이썬 리스트 컴프리헨션 문법

왼쪽에서 오른쪽으로 읽습니다. 표현식은 새 항목이 어떤 모양인지, for 부분은 항목이 어디서 오는지 알려 줍니다.

squares = [x * x for x in range(6)]
print(squares)          # [0, 1, 4, 9, 16, 25]

names = [" Hi ", "  QA "]
clean = [n.strip().lower() for n in names]
print(clean)            # ['hi', 'qa']

같은 결과를 반복문으로 만들면 세 줄이 더 필요합니다. 한 줄이 읽기 쉬운 상태로 유지되는 한, 이 간결함이 핵심입니다.

조건으로 항목 걸러내기

반복문 뒤에 if를 붙이면 조건을 통과한 항목만 남습니다.

evens = [x for x in range(10) if x % 2 == 0]
print(evens)            # [0, 2, 4, 6, 8]

if-else를 쓴 파이썬 리스트 컴프리헨션

모든 항목에 대해 값을 만들고 싶다면 if-else를 for 앞에 두세요. 끝에 else를 두면 문법 오류가 되기 때문에 초보자가 가장 많이 틀리는 부분입니다.

labels = ["even" if x % 2 == 0 else "odd" for x in range(4)]
print(labels)           # ['even', 'odd', 'even', 'odd']

중첩 반복문과 평탄화

여러 for 절은 중첩 반복문과 같은 순서로 읽습니다. 리스트의 리스트를 평탄화하거나 쌍을 만들 때 쓰세요.

matrix = [[1, 2, 3], [4, 5, 6]]
flat = [n for row in matrix for n in row]
print(flat)             # [1, 2, 3, 4, 5, 6]

pairs = [(x, y) for x in range(1, 3) for y in "ab"]
print(pairs)            # [(1, 'a'), (1, 'b'), (2, 'a'), (2, 'b')]

행렬을 전치하려면 컴프리헨션 안에 다른 컴프리헨션을 넣습니다. [[r[i] for r in matrix] for i in range(3)]는 [[1, 4], [2, 5], [3, 6]]을 만듭니다. 가독성이 떨어지기 전의 현실적인 한계는 두 단계입니다.

딕셔너리, 집합, 제너레이터 형태

괄호만 바꾸면 같은 아이디어로 다른 자료형을 만들 수 있습니다.

lengths = {w: len(w) for w in ["pip", "venv", "uv"]}   # {'pip': 3, 'venv': 4, 'uv': 2}
unique  = {len(w) for w in ["pip", "venv", "uv", "git"]}  # {2, 3, 4}
total   = sum(x * x for x in range(1_000_000))           # generator, no list

자료형 자체의 비교는 리스트, 튜플, 집합, 딕셔너리 가이드(영문)에서 확인하세요.

속도와 메모리: 저희가 측정한 결과

Python 3.13.6에서 정수 100,000개로 각 방식을 측정하고, 다섯 번 실행 중 가장 좋은 값을 사용했습니다. 숫자는 환경마다 다르지만 순서는 일정합니다.

컴프리헨션은 append 반복문보다 분명히 빠르지만, 시간 대부분이 조건 검사에 쓰이면 차이가 줄어듭니다. 메모리 면에서는 2배로 만든 정수 100만 개의 리스트가 약 8.4 MB였고, 같은 내용의 제너레이터 객체는 200바이트였습니다. 값을 하나씩 생성하기 때문입니다.

리스트 컴프리헨션을 쓰지 말아야 할 때

• 출력이나 저장 같은 부수 효과만 필요할 때. 일반 반복문을 쓰세요. 컴프리헨션은 쓸모없는 리스트를 만듭니다.
• 로직이 여러 단계일 때. 두 번째 변수나 try가 필요하면 반복문이나 작은 함수를 작성하세요.
• 데이터가 매우 클 때. 제너레이터 표현식을 쓰면 파이썬이 전체 리스트를 메모리에 두지 않습니다.
• for 절이 두 개를 넘을 때. 다음에 읽을 사람을 위해 이름 있는 단계로 나누세요.

마무리

짧은 한 줄이 의도를 정확히 전달할 때는 파이썬 리스트 컴프리헨션을, 그렇지 않을 때는 반복문을 쓰세요. 기본 형태로 시작해 필터를 하나 더하고, 순서가 자연스러워질 때까지 if-else 형태를 연습해 보세요.

자주 묻는 질문

더 배우기 (영문 글)