[Python] 리스트 내포 if / else 구문 사용 방법

파이썬에서 내포는 변환(map) / 필터링(filter) 작업을 동시에 할 수 있는데,

 

리스트 내포를 활용할 때, 조건에 해당하는 것만 출력할 때는 [ ① 표현식 ② for ③ if ] 순으로 작성하면 된다.

 

[예시 1] 0~9 범위 내에서 짝수인 경우만 제곱해서 리스트 출력

[ x**2 for x in range(10) if x%2==0 ]
# [0, 4, 16, 36, 64]

 

if / else 조건 모두 활용하는 경우 [ ① (if 조건일 때) 표현식 ② if ③ else + (else 조건일 때) 표현식 ④ for ] 순으로 작성하면 된다.

 

[예시 2] 0~9 범위 내에서 짝수인 경우만 제곱하고, 홀수는 세제곱하여 리스트 출력

[ x**2 if x%2==0 else x**3 for x in range(10) ]
# [0, 1, 4, 27, 16, 125, 36, 343, 64, 729]