[파이썬] 리스트의 컴프리헨션 활용 예시

리스트 컴프리헨션은 파이썬에서 간결하게 리스트를 생성하는 방법입니다. 이를 활용하여 여러 가지 작업을 수행할 수 있습니다. 아래에는 몇 가지 리스트 컴프리헨션 예시를 소개하겠습니다.

1. 기존 리스트에서 새로운 리스트 생성하기

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

위의 예시에서는 numbers 리스트의 각 요소를 제곱하여 squared_numbers 리스트를 생성합니다. 결과는 [1, 4, 9, 16, 25]가 됩니다.

2. 조건을 만족하는 요소만 포함하는 리스트 생성하기

numbers = [1, 2, 3, 4, 5]
even_numbers = [x for x in numbers if x % 2 == 0]

위의 예시에서는 numbers 리스트의 요소 중에서 짝수인 요소만을 선택하여 even_numbers 리스트를 생성합니다. 결과는 [2, 4]가 됩니다.

3. 중첩된 리스트의 평면화

nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened_list = [x for sublist in nested_list for x in sublist]

위의 예시에서는 nested_list와 같이 중첩된 리스트를 평면화하여 flattened_list를 생성합니다. 결과는 [1, 2, 3, 4, 5, 6, 7, 8, 9]가 됩니다.

리스트 컴프리헨션은 이 외에도 다양한 활용 예시가 있으며, 자유롭게 활용할 수 있습니다. 파이썬의 강력한 기능 중 하나인 리스트 컴프리헨션을 효과적으로 활용하여 코드를 간결하고 읽기 쉽게 만들어보세요!