Python - List Comprehension

List Comprehension의 사용 방법

파이썬 프로그래밍 언어에서 지원되는 기능 중 하나로, 리스트를 생성하고 초기화하기 위한 효율적인 방법중 하나이다.

그렇게 어렵지 않으니 바로 예제코드를 보면서 알아보자.

array = []

for i in range(10):
	
    if ( i % 2 == 0 ):
		array.append(i)
        
# array = [0,2,4,6,8]
	

이 코드는 0~9 까지 짝수인 숫자를 array 리스트에 넣는 코드이다. 보통 리스트에 어떠한 값을 조건에 따져서 넣을때, 반복문과 조건문을 통하여 위 그림과 같은 형태로 구현이 된다.

하지만 아래 그림처럼 List Comprehension 으로 작성한다면 한줄로 작성이 가능하다.

array = [ i for i in range(10) if i % 2 == 0 ]

# array = [0,2,4,6,8]

이와 그림처럼 파이썬은 3~4줄 짜리 코드를 하나의 문장으로 작성할 수 있는 기능을 가졌다. 짧다고 해서, 과도하게 사용한다면 가독성이 떨어질 수 있으니 이점을 주의하고 사용하도록 하자.