[Python] 리스트내포

리스트내포; list 안에 for , if 사용하는 문법

형식1) 변수=[실행문 for 변수 in 열거형객체]

x=[1,2,3,4,5]
print (x**2)
# <출력> TypeError: unsupported operand type(s) for ** or pow(): 'list' and 'int'

lst = [y**2 for y in x]
print(lst)
# <출력> [1, 4, 9, 16, 25]

직접 곱하면 Error 발생 하여 x객체의 원소를 하나씩 y변수에 넣어서 **2 해주어야 한다.

 

 

형식2) 변수=[실행문 for 변수 in 열거형객체 if 조건문]  ; 조건문이 true 일때

ex)

1~10까지의 수에서 2의 배수 추출

lst = list(range(1,11))             # 1~10까지의 수
lst2=[i*2 for i in lst if i%2==0]   # 2의 배수 추출 > 조건문 i%2==0 / 변수=i
print(lst2)
# <출력> [4, 8, 12, 16, 20]

 

형식3) 변수=[값1 if 조건 else 값2 for 변수 in 열거형 객체]   ; 조건문이 false 일때