파이썬 리스트 컴프리헨션

리스트를 사용할 때 내 지식으로 작성했던 반복문 코드는 여러줄인데, 프로그래머스에서 다른 사람들의 문제 풀이들을 보다보면 한줄의 코드로 압축되어 작성한 것을 볼 수 있었다.

a = []
for i in range (5):
	a.append(i)
    
=> [0,1,2,3,4]

a = [ i for i in range(5) ]

=> [0,1,2,3,4]

위와 같은 형태를 리스트 컴프리헨션 이라고 한다.
대괄호 안에 반복문과 조건문을 이용하여 리스트를 초기화 하는 것이다.

a = [ i for i in range(20) if i%2 == 1 ]
=> [1,3,5,7,9,11,13,15,17,19]

반복문과 조건문을 함께 사용한 예제이다.
0~19 중 홀수만 리스트 a 에 초기화 한다.

a = [ i * i for i in range(1,10) ]
=> [1,4,9,16,25,36,49,64,81]

이렇게 거듭제곱의 값을 리스트에 초기화 할 수도 있다.

리스트 컴프리헨션은 2차원 리스트 초기화시에 유용하게 쓰인다.

a = [[0]*M for _ in range(N)]

위 코드는NxM 크기의 2차원 리스트를 초기화 하는 코드다.
N 행, M 열 의 크기를 가진 2차원 리스트가 초기값 0 으로 초기화 되게 된다.

리스트 컴프리헨션을 통해 더 가독성이 높은 코드를 작성 할 수 있을 것 같다.