python: for 문 if 문 한 줄 코딩 하는법

파이썬 이용시 코드를 깔끔하게 하기 위해

for 문 과 if 문 그리고 이 둘을 동시에 python 코드를 한줄로 코딩하고 싶을 때가 있다.

각각 문법은 아래와 같다.

1. for 문

sample = [1,2,3]

# 한줄 코드
[i for i in sample]

# 원래 코드
for i in sample:
    print(i)

 

2. if 문

sample = 3

if sample > 2: print(True)

 

3. if else 문

if 앞에 if 시 발생한 조건을 적는다.

sample = 3

True if sample > 2 else False

 

4. for문 + if 문

for 문 뒤에 조건을 적는다.

sample = [1,2,3]

[i for i in sample if i > 2]

 

5. for문 + if else 문

for 문 앞에 조건을 적는다.

sample = [1,2,3]

[True if i > 2 else False for i in sample ]