파이썬 리스트 내포(list comprehension) 실전 예제

파이썬 리스트 내포의 이해

리스트 내포(List comprehension)는 파이썬 프로그래밍에서 강력한 기능 중 하나로, 리스트를 간결하게 생성하고 조작하는 방법을 제공합니다. 이 기법은 특히 반복문을 사용하여 리스트의 각 요소를 변환하거나 필터링할 때 유용하게 활용됩니다. 본 포스팅에서는 리스트 내포의 기본 개념과 다양한 활용 사례를 살펴보도록 하겠습니다.

리스트 내포의 기본 구조

리스트 내포는 아래의 구문 형태로 작동합니다:

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

여기서 각 요소는 다음과 같은 역할을 합니다:

• 표현식: 리스트에 추가될 각 요소를 생성하기 위한 식입니다.
• 변수: 반복가능 객체에서 각각의 요소를 가리킵니다.
• 반복가능 객체: 리스트, 튜플, 문자열 등 순회할 수 있는 객체를 의미합니다.
• 조건식: 선택적으로 추가될 수 있는 부분으로, 조건에 만족하는 요소만 리스트에 포함됩니다.

기본 예제: 단순 리스트 생성하기

리스트 내포를 사용하여 1부터 5까지의 숫자를 제곱한 리스트를 만들어 보겠습니다:

squares = [x ** 2 for x in range(1, 6)]

이 코드는 squares 변수에 [1, 4, 9, 16, 25]를 담습니다. 기존의 for 반복문을 사용하는 것보다 훨씬 간결하게 작성할 수 있습니다.

리스트 내포의 다양한 활용 방법

리스트 내포는 단순히 리스트를 생성하는 것 외에도 다양한 방식으로 활용될 수 있습니다. 몇 가지 예를 소개하겠습니다.

필터링하기

리스트 내포를 이용하여 특정 조건을 만족하는 요소만을 선택할 수 있습니다. 예를 들어, 짝수인 숫자만 포함된 리스트를 생성하는 방법은 다음과 같습니다:

even_numbers = [x for x in range(10) if x % 2 == 0]

이 코드는 even_numbers 변수에 [0, 2, 4, 6, 8]을 저장하게 됩니다.

조건부 표현식 사용하기

리스트 내포에서 조건부 표현식을 사용할 수 있습니다. 예를 들어, 짝수는 제곱하고 홀수는 세제곱하여 새로운 리스트를 생성할 때는 아래와 같이 작성할 수 있습니다:

transformed_numbers = [x**2 if x % 2 == 0 else x**3 for x in range(10)]

이 경우 transformed_numbers는 [0, 1, 4, 27, 16, 125, 36, 343, 64, 729]의 값을 가집니다.

중첩 리스트 내포

리스트 내포는 중첩하여 사용할 수도 있습니다. 예를 들어, 2D 리스트를 평평하게 만들고 싶다면 다음과 같은 코드를 사용할 수 있습니다:

matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened = [num for row in matrix for num in row]

이 코드는 flattened에 [1, 2, 3, 4, 5, 6, 7, 8, 9]를 저장합니다.

리스트 내포와 가독성

리스트 내포는 간결한 코드 작성을 가능하게 하지만, 가독성을 고려하여 사용해야 합니다. 너무 복잡한 내포는 오히려 이해하기 어렵게 만들 수 있으므로 적절한 수준에서 사용하는 것이 좋습니다.

정리

리스트 내포는 파이썬의 유용하고 강력한 기능으로, 리스트를 간편하게 생성하고 조작할 수 있는 방법을 제공합니다. 이 기능을 이해하고 활용함으로써 코드 작성의 효율성을 높일 수 있습니다. 그러나 항상 가독성을 고려하여 적절한 수준에서 사용하는 것이 중요합니다.

본 포스팅을 통해 리스트 내포의 기본 개념과 활용 방법을 이해하는 데 도움이 되었기를 바랍니다.

자주 찾으시는 질문 FAQ