Python 리스트 컴프리헨션 기초와 사용 예시

파이썬은 현대 프로그래밍 언어 중 하나로, 그 독창성과 표현력 덕분에 많은 개발자들 사이에서 인기를 끌고 있습니다. 이 글에서는 파이썬의 리스트 컴프리헨션(List Comprehension)에 대해 살펴보겠습니다. 이는 리스트를 직관적이고 효율적으로 생성할 수 있는 방법으로, 매우 파이썬스러운(pythonic) 문법으로 인정받고 있습니다.

리스트 컴프리헨션이란?

리스트 컴프리헨션은 반복문을 활용하여 리스트를 생성하는 간결한 방법입니다. 전통적인 방법으로 리스트를 만들기 위해서는 반복문을 사용해 데이터를 하나씩 추가해야 하는데, 리스트 컴프리헨션을 이용하면 이러한 과정을 한 줄의 코드로 간단히 처리할 수 있습니다. 기본적인 문법은 다음과 같습니다.

[expression for item in iterable]

위의 코드에서 expression은 리스트에 포함될 요소를 생성하는 식이며, iterable은 순회할 수 있는 데이터 구조를 의미합니다. 이런 방식으로 리스트를 효율적으로 만들 수 있습니다.

리스트 컴프리헨션의 활용 예

리스트 컴프리헨션은 여러 가지 상황에서 유용하게 사용될 수 있습니다. 예를 들어, 1부터 10까지의 숫자를 포함하는 리스트를 만들고 싶다면, 간단하게 다음과 같이 작성할 수 있습니다.

numbers = [i for i in range(1, 11)]

위의 코드를 실행하면 numbers에는 1부터 10까지의 값이 담긴 리스트가 생성됩니다. 이에 대한 출력 결과는 다음과 같습니다.

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

조건을 통한 리스트 생성

리스트 컴프리헨션을 사용할 때 특정 조건을 부여하여 리스트를 구성할 수도 있습니다. 예를 들어, 짝수만 포함된 리스트를 만들고자 한다면, 다음과 같이 코드를 작성할 수 있습니다.

even_numbers = [x for x in range(1, 11) if x % 2 == 0]

위의 코드를 실행하면 짝수만 포함된 리스트가 생성됩니다. 출력 결과는 다음과 같습니다.

[2, 4, 6, 8, 10]

삼항 연산자를 통한 조건부 리스트 생성

리스트 컴프리헨션은 if문과 else문을 결합하여 삼항 연산자를 사용할 수도 있습니다. 예를 들어, 주어진 목록에서 짝수는 그대로 두고 홀수는 10을 곱한 값을 포함하는 리스트를 만들고 싶다면, 다음과 같이 작성할 수 있습니다.

result = [n * 10 if n % 2 != 0 else n for n in range(1, 11)]

위 코드를 실행하면 결과는 다음과 같습니다.

[10, 2, 30, 4, 50, 6, 70, 8, 90, 10]

중첩 루프를 활용한 리스트 컴프리헨션

리스트 컴프리헨션에서는 중첩된 반복문을 사용할 수도 있습니다. 예를 들어, 두 개의 다른 리스트에서 모든 조합을 생성하고 싶다면, 다음과 같이 작성할 수 있습니다.

combinations = [(x, y) for x in ['A', 'B', 'C'] for y in ['1', '2', '3']]

이 코드를 실행하면 다음과 같은 조합 리스트가 생성됩니다.

[('A', '1'), ('A', '2'), ('A', '3'), ('B', '1'), ('B', '2'), ('B', '3'), ('C', '1'), ('C', '2'), ('C', '3')]

마무리

리스트 컴프리헨션은 데이터 필터링과 변형을 간단하면서도 효과적으로 수행할 수 있는 방법입니다. 간단한 조건을 통해 데이터 가공을 수행할 수 있기 때문에 코드의 가독성을 높일 수 있으며, 실행 속도 또한 향상시킬 수 있습니다. 하지만 복잡한 조건이나 비즈니스 로직이 필요할 경우, 리스트 컴프리헨션보다는 전통적인 반복문을 사용하는 것이 더 명확할 수 있습니다.

이러한 리스트 컴프리헨션의 특징을 이해하고 활용하면, 파이썬으로 작성하는 코드의 효율성과 가독성을 더욱 높일 수 있습니다. 다양한 예제를 통해 실력을 쌓고, 더 나아가 복잡한 문제를 해결하는 데에도 도움을 받을 수 있습니다.

자주 묻는 질문 FAQ