리스트 컴프리헨션 (List Comprehension)

리스트 컴프리헨션(List Comprehension)은 파이썬에서 새로운 리스트를 간결하고 선언적인 방식으로 생성할 수 있도록 해주는 강력한 기능입니다. 반복문(for 루프)과 조건문(if)을 사용하여 리스트를 만드는 과정을 한 줄의 코드로 줄여줍니다.

 

기본 구조는 다음과 같습니다.

[표현식 for 항목 in 순회_가능한_객체]

예를 들어 [1, 2, 3, 4, 5] 리스트를 입력받아 각 요소를 2배로 하여 [2,4,6,8,10] 을 출력한다고 해 봅시다.

이때 리스트 컴프리헨션을 사용한다면 다음과 같습니다.

def solution(numbers):
    answer = [num*2 for num in numbers]
    return answer

 

또한, 리스트 컴프리헨션은 조건을 추가하여 특정 조건을 만족하는 요소를 가져올 수 있습니다.

예를 들어 문자열 요소를 담은 words 라는 리스트에서 문자열의 길이가 5 이상인 요소만 뽑아 리스트를 만든다고 한다면

다음과 같이 작성할 수 있습니다. 

words = ["apple", "banana", "kiwi", "orange", "grape"]
long_words = [word for word in words if len(word) >= 5]
print(long_words)  # 출력: ['apple', 'banana', 'orange', 'grape']