[Python] 리스트 컴프리헨션

리스트 컴프리헨션이란?

• 리스트 컴프리헨션은 리스트를 초기화하는 방법 중 하나입니다.
• 리스트 컴프리헨션을 이용하면 대괄호([]) 안에 조건문과 반복문을 넣는 방식으로 리스트를 초기화 할 수 있습니다.

예제 코드

• 간단히 0부터 19까지의 수 중에서 홀수만 포함하는 리스트를 만들고자 할 떄는 다음과 같이 리스트를 선언할 수 있습니다.
• 아래 예제 코드만 봐도, 리스트 컴프리헨션으로 표현 방법이 훨씬 소스코드가 짧고 간결한 것을 볼 수 있습니다.

일반 적인 표현

list = []
for i in range(20):
    if(i % 2 == 1):
        list.append(i)

print(list)

리스트 컴프리헨션 표현

array = [i for i in range(20) if i % 2 == 1]

print(array)

2차원 리스트, 컴프리헨션 포현으로 만들기

• 리스트 컴프리헨션 표현을 통해 2차원 리스트를 초기화할 때 매우 효과적으로 사용할 수 있습니다.

# n * M 크기의 2차원 리스트 초기화
n = 3
m = 4

array = [[0] * m for _ in range(n)]