리스트 컴프리헨션(List Comprehension)

리스트 컴프리헨션은 간결하고 효율적인 리스트 생성 방법이다. 반복문과 조건문을 사용하여 리스트를 생성하는 과정을 한 줄로 표현할 수 있다.

 

# 19까지의 숫자 중에서 홀수만 포함하는 리스트 만들기

# 일반적인 방법
array = []

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

print(array)

# 리스트 컴프리헨션
array = [i for i in range(20) if i % 2 == 1]

print(array)

출력 결과 >>
[1, 3, 5, 7, 9, 11, 13, 15, 17, 19]

---

# N x M 크기(3행 4열)의 2차원 리스트 초기화하기

# 일반적인 방법
n = 3
m = 4
array = []

for _ in range(n):
	a = [0] * m
    array.append(a)
 
print(array)

# 리스트 컴프리헨션
n = 3
m = 4
array = [[0]*m for _ in range(n)]

print(array)

출력 결과>>
[[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0]]

---

# numbers 리스트의 요소를 제곱하여 새로운 리스트 만들기

# 일반적인 방법
numbers = [1, 2, 3, 4, 5]
a = []

for num in numbers :
	a.append(num**2)

print(a)

# 리스트 컴프리헨션
numbers = [1, 2, 3, 4, 5]
a = [num**2 for num in numbers]

print(a)

출력 결과>>
[1, 4, 9, 16, 25]