Python - 지능형 리스트(List Comprehension) 이해

오늘은 Python에서 많이 사용되는 지능형 리스트(List Comprehension)에 대해서 이해해보는 시간을 가져보고자한다.

지능형 리스트는 List 값을 생성하는데 코드를 보다 간결하게 작성할 수 있도록 도와준다.

sum = []
for number in range(10):
	sum.append(number)

print(sum)
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

위 코드를 지능형 리스트로 만들면 아래와 같다.

sum = [number for number in range(10)]

print(sum)
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

 

지능형 리스트는 오늘쪽으로 계산식이 넘어가고, 해당 결과 값이 가장 앞 변수로 선언하게 된다. 그리고 파이썬에서 리스트를 의미하는 대괄호[]를 통해 리스트 형식으로 생성된다. 그림으로 표현하면 다음과 같이 표현이 가능하다.

아래와 같이 조건을 하나 더 넣어서, 6보다 큰 수를 List로 만드는 간단한 예제가 있다.

sum = []
for number in range(10):
	if number > 5:
		sum.append(number)

print(sum)
[6, 7, 8, 9]

이를 지능형으로 표현한다면 아래와 같다.

sum = [number for number in range(10) if number > 5]

print(sum)
[6, 7, 8, 9]

조건을 계속 더해 나가는 식으로 표현이 가능하다.