[python] 리스트 컴프리헨션 - 리스트를 한줄로 초기화 시키는 법

리스트 컴프리헨션 이란 ?

---

리스트를 초기화 시키는 방법 중 하나이다

리스트 안에다가 조건문과 반복문을 넣어서 초기화를 시켜주는 방법이다

 

알고리즘 문제에서 아주 자주 등장하니까 꼭꼭 능숙하게 사용할 수 있어야한다 !

 

 

사용 예시

---

 

1. 0-9까지의 배열

lst0 = [i for i in range(10)]
print(lst0)

>>[0, 1, 2, 3, 4, 5, 6, 7, 8]

 

2. 0-19 까지의 홀수만 있는 배열

lst1 = [i for i in range(20) if i%2 != 1]
print(lst1)

>>[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

 

3. i*i (제곱 값) 을 나타내는 배열

lst2 = [i*i for i in range(9)]
print(lst2)

>>[0, 1, 4, 9, 16, 25, 36, 49, 64]

 

4. N*M 의 크기의 2차원 리스트

n = 3
m = 2
lst3 = [[0]*m for i in range(n)]
print(lst3)


>>[[0, 0], [0, 0], [0, 0]]