[Python] List comprehension 리스트 안에서 for문, if문 사용하기

 

파이썬은 리스트 안에 for문(반복문)과 if(조건문)을 사용할 수 있다.

리스트 안에 식, 반복문, 조건문 등을 지정하여 리스트를 생성하는 것을

리스트 컴프리헨션(list comprehension)이라고 한다.

 

1~9까지의 정수를 담고있는 리스트를 생성하는 기본적인 방법

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9]

 

일반적인 for문

arr = []
for i in range(1, 10):
	arr.append(i)

list안에서 for문을 사용하는 방법

arr = [i for i in range(1, 10)]

 

list안에서 for문, if문을 모두 사용하는 방법 (1~9 사이의 홀수 리스트 생성)

arr = [i for i in range(1, 10) if (i%2 == 1)]

 

 

 

위와 같이 list comprehension을 이용하면  코드를 훨씬 더 간결하게 작성할 수 있다.