[파이썬] 리스트 컴프리헨션의 사용법

리스트 컴프리헨션(list comprehension)은 파이썬에서 매우 유용한 기능 중 하나입니다. 리스트 컴프리헨션을 사용하면 간단하고 간결한 방식으로 리스트를 생성할 수 있습니다. 이번 블로그 포스트에서는 리스트 컴프리헨션의 사용법을 알아보겠습니다.

기본 구문

리스트 컴프리헨션은 다음과 같은 기본 구문으로 작성됩니다:

new_list = [expression for item in iterable]

위의 코드에서 expression은 각 item에 대한 표현식(expression)입니다. item은 리스트를 생성하는 데 사용될 변수이며, iterable은 순회 가능한(iterable) 객체입니다.

예제 코드

다음의 예제 코드를 통해 리스트 컴프리헨션의 사용법을 자세히 알아보겠습니다:

# 1부터 10까지의 숫자로 이루어진 리스트 생성
numbers = [x for x in range(1, 11)]
print(numbers)

# 리스트 내 각 요소의 제곱 값으로 이루어진 리스트 생성
squared_numbers = [x ** 2 for x in numbers]
print(squared_numbers)

# 문자열 리스트에서 길이가 5 이상인 문자열만 선택하여 새로운 리스트 생성
fruits = ["apple", "banana", "orange", "kiwi", "mango", "pineapple"]
long_fruits = [fruit for fruit in fruits if len(fruit) >= 5]
print(long_fruits)

위의 코드를 실행하면 다음과 같은 결과가 출력됩니다:

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
['banana', 'orange', 'mango', 'pineapple']

조건문 활용

리스트 컴프리헨션에서는 조건문을 활용하여 특정한 조건을 만족하는 요소만 선택하는 것도 가능합니다. 위의 예제 코드에서 fruits 리스트에서 문자열 길이가 5 이상인 요소만 선택하는 부분이 그 예입니다.

조건문은 리스트 컴프리헨션의 마지막에 추가할 수 있고, 필요에 따라 여러 개의 조건을 연결하여 사용할 수도 있습니다.

결론

리스트 컴프리헨션은 파이썬에서 리스트를 생성하는 간편하고 간결한 방법입니다. 특히 순회 가능한 객체를 통해 리스트를 생성하거나, 특정한 조건을 만족하는 요소만 선택하고 싶을 때 유용하게 사용할 수 있습니다. 위의 예제 코드를 통해 리스트 컴프리헨션의 기본 사용법을 알아보았습니다. 여러분도 실제 프로젝트에서 리스트 컴프리헨션을 사용하여 효율적이고 가독성 좋은 코드를 작성해보세요!