[파이썬] 리스트 컴프리헨션

파이썬에서 컴프리헨션(Comprehension)은 자료구조(list, dictionary, set)에 데이터를 좀 더 쉽고 간결하게 담기 위한 문법이다. 그 중에서 리스트를 생성하는 컴프리헨션인 '리스트컴프리헨션'을 알아보자.

 

일반적으로 리스트를 생성하는 코드는 다음과 같다. 

numbers = []
for n in range(1, 10+1):
    numbers.append(n)

 

하지만 컴프리헨션으로 표기하면 다음과 같이 간결하게 표기할 수 있다. 

[x for x in range(10)]

 

컴프리헨션은 if 키워드를 지원한다. 예를 들어 짝수를 담는 리스트컴프리헨션은 다음과 같이 작성할 수 있다. 

[x for x in range(1, 10+1) if x % 2 == 0]

[2, 4, 6, 8, 10]