파이썬 리스트 컴프리헨션 심화 예제 코드

파이썬에서 리스트를 생성하고 조작하는 과정은 개발자에게 매우 중요한 기술입니다. 그 중에서도 리스트 컴프리헨션은 리스트를 직관적이고 간결하게 만드는 방법으로, 파이썬의 문법 중 하나입니다. 이번 포스트에서는 리스트 컴프리헨션의 기초부터 다양한 응용 방식까지 자세히 알아보겠습니다.

리스트 컴프리헨션이란?

리스트 컴프리헨션은 파이썬에서 리스트를 생성할 때 반복문과 조건문을 결합하여 보다 효율적으로 값을 만들 수 있는 문법입니다. 이 방식은 대괄호([])로 감싸인 표현식을 사용하여 리스트를 쉽게 생성할 수 있도록 도와줍니다. 전통적인 방법으로 리스트를 만드는 것보다 훨씬 간결하고 읽기 쉬운 코드를 작성할 수 있습니다.

기초 문법

리스트 컴프리헨션의 기본 문법은 다음과 같이 표현됩니다:

리스트 = [표현식 for 변수 in 반복 가능한 객체]

예를 들어 0부터 4까지의 수를 담은 리스트를 생성하고자 할 때, 전통적인 방법으로는 여러 줄의 코드가 필요합니다:

리스트 = []
for i in range(5):
  리스트.append(i)

하지만 리스트 컴프리헨션을 사용하면 단 한 줄로 간단하게 만들 수 있습니다:

리스트 = [i for i in range(5)]

이 코드로 생성된 리스트는 [0, 1, 2, 3, 4]가 됩니다.

리스트 컴프리헨션의 응용

리스트 컴프리헨션은 단순히 리스트를 생성하는 것 외에도 다양한 방식으로 활용할 수 있습니다.

1. 수식이나 함수 적용하기

리스트 컴프리헨션의 가장 큰 장점 중 하나는 수식을 쉽게 적용할 수 있다는 점입니다. 아래는 각 요소에 10을 곱하는 예제입니다:

결과 = [i * 10 for i in range(5)]

이 코드를 실행하면 [0, 10, 20, 30, 40]의 리스트가 생성됩니다. 또한 사용자 정의 함수를 적용하는 것도 가능합니다.

def 변환(x):
  return str(x) + "번"
결과 = [변환(i) for i in range(5)]

위 코드는 [‘0번’, ‘1번’, ‘2번’, ‘3번’, ‘4번’] 의 결과를 도출합니다.

2. 조건문 활용하기

리스트 컴프리헨션에서는 조건문을 사용하여 필터링할 수 있습니다. 예를 들어, 짝수만 추출하는 예제는 다음과 같습니다:

짝수 = [i for i in range(10) if i % 2 == 0]

이 코드를 통해 [0, 2, 4, 6, 8]의 리스트를 얻을 수 있습니다.

3. if-else 문 사용하기

리스트 컴프리헨션에서 if-else 구문을 활용하여 특정 조건에 따라 다른 값을 반환할 수도 있습니다. 예를 들어, 0부터 4까지의 수를 체크하여 짝수는 그대로 두고 홀수는 ‘홀수’라는 문자열로 대체하는 경우:

결과 = [i if i % 2 == 0 else '홀수' for i in range(5)]

결과는 [0, ‘홀수’, 2, ‘홀수’, 4]로 나타납니다.

4. 중첩 for문 사용하기

리스트 컴프리헨션은 중첩된 반복문을 표현하는 데도 유용합니다. 예를 들어, (i, j) 형식의 조합을 만들고 싶다면:

조합 = [(i, j) for i in range(2) for j in range(3)]

이 코드는 [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2)]의 결과를 생성합니다.

리스트 컴프리헨션의 장점과 단점

리스트 컴프리헨션의 가장 큰 장점은 코드가 간결해지고 가독성이 향상된다는 것입니다. 또한, 속도 면에서도 일반적인 for문보다 빠르게 동작하는 경우가 많습니다. 그러나 복잡한 조건문이나 중첩 loop가 필요할 경우, 코드의 가독성이 오히려 떨어질 수 있으므로 그 경우에는 전통적인 방법을 사용하는 것이 좋습니다.

마치며

리스트 컴프리헨션은 파이썬에서 리스트를 효율적으로 다룰 수 있는 매우 유용한 도구입니다. 여러 가지 방법으로 활용할 수 있으며, 특히 수식, 조건문과의 조합은 다양한 프로그래밍 상황에서 큰 도움이 됩니다. 이 포스트를 통해 리스트 컴프리헨션의 기초와 응용 방법을 이해하셨기를 바랍니다. 파이썬을 활용하여 더욱 다양한 프로젝트를 시도해보세요!

자주 찾는 질문 Q&A