파이썬 List comprehension

리스트를 생성할 때 comprehension을 사용하면 간결하게 만들 수 있다. 

 

comprehension의 형태

 

[ x ** 2 for x in num if x>0 ]

[ output | for 변수 in 범위  | 조건문]

조건문은 optional이다. 있어도 되고 없어도 된다.

*** output에 있는 변수는 꼭 뒤의 for문에 사용한 변수와 같아야 한다.

위의 예시는 num이라는 변수 중 0보다 큰 수들의 제곱 형태를 원소로 가지는 리스트를 만드는 것이다.

 

 

예시

#1. 
fruits = ["banana", "orange", "melon", "apple", "watermelon"]
print([w for w in fruits if len(w)>5])
--> ["banana", "orange", "watermelon"]

#2. 
numbers = [1,7,8,2,4,5]
print(sorted([x+2 for x in numbers if x>=4 and x<=7])
--> [6,7,9]

 

 

 

중첩 list comprehension

 

[ output | 바깥 for문 안 for문]