[python] 리스트 컴프리헨션(List Comprehension) : 반복문과 조건문 줄이고 싶을 때

 

 

리스트 컴프리헨션이란?

 

 

파이썬에서 간결하게 반복문과 조건문을 줄이는 것!

 

아래와 같이 사용한다

list = [ 요소에 연산 수행 for interable의 개별 요소 in iterable if 조건]


iterable : 반복 가능한 객체

 

 

 

 

 

 

반복문 사용 예시

 

 

1에서 10까지 2를 곱한 리스트 생성하는 예시

 

a_list = [ i*2 for i in range(1,11) ]

print(a_list)
#[2,4,6,8,10,12,14,16,18,20]

 

 

 

 

 

 

조건문 사용 예시

 

 

1에서 10까지 중 홀수만 수 출력 예시

 

 

odd = [i for i in range(1, 11) if i % 2 == 1]

print(odd) 
#[1, 3, 5, 7, 9]

 

 

 

 

 

다음과 같이 리스트가 아닌 집합, 딕셔너리 형태로도 출력 가능하다

 

집합 = { i for i in range(5) }

print(집합)
#{0,1,2,3,4}


딕셔너리 = { i:i for i in range(5) }

print(딕셔너리)
#{0:0, 1:1, 2:2, 3:3, 4:4}