#6 TIL Python - List Comprehension

리스트 컴프리헨션(List Comprehension)은 파이썬에서 리스트를 간결하고 효율적으로 생성할 수 있는 문법이다. 기존의 for 루프를 사용한 리스트 생성보다 더 직관적이고 간결한 코드를 작성할 수 있게 해준다.

기본 문법

리스트 컴프리헨션의 기본 문법은 다음과 같다

[표현식 for 항목 in 반복 가능 객체 if 조건]

이 문법을 활용하면 반복 가능 객체(iterable)의 각 항목에 표현식을 적용하여 새로운 리스트를 생성할 수 있다. 조건문을 추가하면 특정 조건을 만족하는 항목만 리스트에 포함시킬 수 있다.

예제 1: 기본 사용법

다음은 1부터 10까지의 숫자 중에서 짝수만을 포함하는 리스트를 생성하는 예제다.

numbers = [1, 2, 3, 4, 5]
squared_numbers = [n ** 2 for n in numbers]
print(squared_numbers)  # 출력: [1, 4, 9, 16, 25]

예제 3: 중첩된 리스트 컴프리헨션

리스트 컴프리헨션은 중첩해서 사용할 수도 있다. 예를 들어, 2차원 리스트(행렬)의 각 요소를 제곱하는 경우는 다음과 같다.

matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]
squared_matrix = [[n ** 2 for n in row] for row in matrix]
print(squared_matrix)
# 출력:
# [
#     [1, 4, 9],
#     [16, 25, 36],
#     [49, 64, 81]
# ]

결론

리스트 컴프리헨션은 파이썬에서 리스트를 간결하고 효율적으로 생성할 수 있는 강력한 도구다. 이를 활용하면 보다 읽기 쉽고 유지보수하기 쉬운 코드를 작성할 수 있다. 다양한 상황에서 리스트 컴프리헨션을 활용하여 코드의 가독성과 효율성을 높일 수 있다.