[python] 리스트 컴프리헨션(List Comprehension)

기본 구조

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

[expression for item in iterable if condition]

 

 

 

• expression: 새 리스트의 각 요소에 대한 식이며, 이는 item을 기반으로 계산됩니다.
• item: iterable에서 각 요소를 나타냅니다.
• iterable: 순회 가능한 데이터 소스로부터 요소들이 추출됩니다.
• if condition: (선택적) 이 조건을 만족하는 item만 expression을 계산하여 새 리스트에 포함됩니다.

 

 

 

예시

기본 사용 예:

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

 

조건 포함 예:

# 0부터 9까지의 숫자 중에서 짝수만의 제곱을 계산하여 리스트 생성
even_squares = [x**2 for x in range(10) if x % 2 == 0]
print(even_squares)  # 출력: [0, 4, 16, 36, 64]