리스트 컴프리헨션(List Comprehension)이란?

리스트 컴프리헨션(List Comprehension)이란?

리스트 컴프리헨션은 파이썬에서 제공하는 강력하고 간결한 기능으로, 기존 리스트를 기반으로 새로운 리스트를 생성할 수 있게 해줍니다. 

리스트 컴프리헨션을 사용하면 더 짧고 읽기 쉬운 코드를 작성할 수 있습니다.

리스트 컴프리헨션의 기본 형식

리스트 컴프리헨션은 다음과 같은 형식을 가집니다:

[표현식 for 아이템 in 반복가능한 객체]

• 표현식: 새로운 리스트의 각 요소를 계산하는 방식입니다.
• 아이템: 반복 가능한 객체의 각 요소입니다.
• 반복가능한 객체: 리스트, 문자열, 튜플 등 반복 가능한 객체입니다.

예시 1: 기본 리스트 컴프리헨션

주어진 리스트의 각 요소를 제곱한 값을 가지는 새로운 리스트를 만들어보겠습니다.

일반적인 방법

numbers = [1, 2, 3, 4, 5]
squared_numbers = []

for num in numbers:
   squared_numbers.append(num ** 2)

print(squared_numbers)  # 출력: [1, 4, 9, 16, 25]

위 내용은 [1,2,3,4,5] 를 순회하며 [1^2, 2^2, 3^2, 4^2, 5^2] 인 리스트를 출력하는 코드인데요 

이런 내용을 파이썬에서는 한 줄로 표현할 수 있어요!

위 내용을 리스트 컴프리헨션으로 표현한다면?

numbers = [1, 2, 3, 4, 5]
squared_numbers = [num ** 2 for num in numbers]

print(squared_numbers)  # 출력: [1, 4, 9, 16, 25]

예시 2: 조건을 포함한 리스트 컴프리헨션

리스트 컴프리헨션에는 조건문도 포함할 수 있습니다. 

조건문을 사용하여 특정 조건을 만족하는 요소만을 선택할 수 있습니다.

[짝수만 선택하는 예시]

일반적인 방법

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

for num in numbers:
   if num % 2 == 0:
       even_numbers.append(num)

print(even_numbers)  # 출력: [2, 4, 6, 8, 10]

위 내용은 [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 를 순회하며 짝수만 포함한 리스트를 출력하는 코드인데요 

이 내용도 파이썬에서는 한 줄로 표현할 수 있어요!

위 내용을 리스트 컴프리헨션으로 표현한다면?

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
even_numbers = [num for num in numbers if num % 2 == 0]

print(even_numbers)  # 출력: [2, 4, 6, 8, 10]

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

리스트 컴프리헨션을 중첩하여 2차원 리스트를 1차원 리스트로 변환할 수도 있습니다.

2차원 리스트를 1차원 리스트로 변환하는 예시

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

flat_list = [num for row in matrix for num in row]

print(flat_list)  # 출력: [1, 2, 3, 4, 5, 6, 7, 8, 9]

신기하죠?

리스트 컴프리헨션의 장점

1. 간결성: 코드가 더 짧고 간결해집니다.
2. 가독성: 코드의 가독성이 향상됩니다.
3. 성능: 일반적으로 for 루프보다 성능이 좋습니다.

요약

• 리스트 컴프리헨션: 기존 리스트를 기반으로 새로운 리스트를 생성하는 간결한 방법.
• 기본 형식: [표현식 for 아이템 in 반복가능한 객체]
• 조건 포함 형식: [표현식 for 아이템 in 반복가능한 객체 if 조건]
• 중첩 리스트 컴프리헨션: 중첩된 리스트도 간단하게 처리할 수 있습니다.

이제 리스트 컴프리헨션에 대해 잘 이해할 수 있겠죠? 

리스트 컴프리헨션은 파이썬에서 매우 유용한 기능으로, 더 효율적이고 깔끔한 코드를 작성하는 데 도움이 됩니다!