[08.17] List Comprehension

리스트 컴프리헨션

• 리스트를 초기화하는 방법 중에 하나이다.
• 대괄호 안에 조건문과 반복문을 넣는 방식으로 리스트를 초기화할 수 있다.

ex) 0~19까지 중 홀수만 배열에 포함하는 리스트

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

 

List Comprehension을 사용하지 않았을 때의 코드

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

print(array)

---

 

List Comprehension은 2차원 리스트를 초기화할 때 매우 효과적으로 사용될 수 있다.

# N X M 크기의 2차원 리스트 초기화
n = 3
m = 4
array = [[0] * m for _ in range(n)]
# _은 반복을 수행하되 반복을 위한 변수의 값을 무시하고자 할 때 언더바(_)를 사용.
print(array)