[파이썬] 컴프리헨션 구문

880
0

리스트, 딕셔너리의 키/값 페어, 집합 처리를 간단하게 만드는 함수를 소개한다.
파이썬에서는 컴프리헨션 이라는 특별한 구문을 사용해 위의 타입들을 간결하게 이터레이션 하며 데이터 구조를 생성할 수 있다.

기존 for 루프를 이용한 리스트 원소 제곱 계산

a = [1,2,3,4,5,6,7,8,9,10]
squares = []
for x in a:
  sqaures.append(x**2)
print(squares)

>>> [1,4,9,16,25,36,49,64,81,100]

컴프리헨션 구문을 사용한 리스트 원소 제곱 계산

a = [1,2,3,4,5,6,7,8,9,10]
squares = [x**2 for x in a]
print(squares)

>>> [1,4,9,16,25,36,49,64,81,100]

컴프리헨션 구문을 사용한 리스트 원소 제곱 계산 응용

a = [1,2,3,4,5,6,7,8,9,10]
even_sqaures = [x ** 2 for x in a if x % 2 == 0]
print(even_squares)

# 리스트의 원소가 제곱된 후 그 값이 나머지가 0이 되는 수
# 또 다른 조건을 추가하며 처리할 수 있다.
>>> [4,16,36,64,100]

답글 남기기

이메일 주소는 공개되지 않습니다.