[파이썬] 리스트 컴프리헨션(List Comprehension) 개념과 예제

리스트 컴프리헨션(List Comprehension)은 파이썬에서 간결하고 효율적으로 리스트를 생성하는 방법 중 하나입니다. 기존의 for 루프를 사용하여 리스트를 생성하는 코드를 한 줄로 간략하게 표현할 수 있도록 도와줍니다.

리스트 컴프리헨션의 구조는 다음과 같습니다:

[표현식 for 항목 in 반복 가능한 객체 if 조건문]

예제를 통해 더 자세히 살펴보겠습니다:

# 0부터 9까지의 제곱 값을 담은 리스트를 생성하는 예제
squares = [x ** 2 for x in range(10)]
print(squares)  # 출력: [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

# 문자열 리스트에서 길이가 5 이하인 단어만 추출하는 예제
words = ["apple", "banana", "grape", "kiwi", "pear", "orange"]
short_words = [word for word in words if len(word) <= 5]
print(short_words)  # 출력: ['apple', 'grape', 'kiwi', 'pear']

# 리스트 내의 숫자를 짝수만 추출하여 새로운 리스트 생성하는 예제
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]

리스트 컴프리헨션을 사용하면 더 간결하게 리스트를 생성하고 데이터를 필터링할 수 있습니다. 다양한 조건을 활용하여 원하는 데이터를 추출하거나 변환하여 새로운 리스트를 만들 수 있습니다.