[Python] 리스트 함축 (List Comprehension)

 

리스트 함축이란

리스트를 수학식과 

유사하게 정의하는 것이다.

 

[expression for i in list]

로 쓸 수 있다.

 

이로 인해

수식과 반복을 간편하게 할 수 있다.

 

예제를 통해 실습해보자.

 

 

 

 

 

1. 한 변의 길이가 1~10인

정사각형의 넓이들을 구하기

 

result1 = [x ** 2 for x in range(1,11)]
 
print(result1)

 

 

  - 1. 결과  

 

 

  - 1. 설명  

for x in range(1, 11) 범위의 x가

x ** 2가 되는 것을 반복하여

result1에 리스트 형식으로

추가된다.

 

 

 

 

 

 

2. 리스트의 길이만 리스트에 추가하기,

리스트 요소의 첫 글자만 리스트에 추가하기

 

list1 = ["This", "Is", "Python"]
 
result2 = [len(x) for x in list1]
result3 = [x[0] for x in list1]
 
print(result2)
print(result3)

 

 

 

  - 2. 결과  

 

 

 

  - 2. 설명  

 

result2 :

list1의 요소를 반복하여

리스트의 형식으로

길이만 출력해낸다.

 

result3 :

list1의 요소를 반복하여

리스트의 형식으로

요소의 첫 글자만 추가한다.

 

 

 

 

 

 

3. 리스트 두 개가 쌍을 이루는

모든 경우의 수 구하기

 

colors = ["white", "black"]
cars = ["bmw5", "sonata", "malibu"]
 
result4 = [ (x , y) for x in colors for y in cars]
 
print(result4)

- 3. 결과  

 

 

 

- 3. 설명

colors의 요소들과

cars의 요소들이 반복되어

(x , y)의 형식을 이루며

리스트 형식으로

추가되어 출력된다.