[Python] 한줄로 for+ if 문 사용하기

파이썬은 for문과 if문을 한 줄로 작성할 수 있다.

for 한줄 작성

a = list()
for i in range(10):
    a.append(i*2)
print(a)

a = [i*2 for i in range(10)]
print(a)

for + if else 한 줄 작성

a=list()
for i in range(10):
    if i%2 == 1:
        a.append(i*2)
    else:
        a.append(i*0)
print(a)

a = [i*2 if i%2==1 else i*0 for i in range(10)]
print(a)

for + if 한줄 작성

else를 쓰지 않는다면 if 문을 뒤로 보내야 한다.

a=list()
for i in range(10):
    if i%2 == 1:
        a.append(i*2)
print(a)

a = [i*2 for i in range(10) if i%2==1]
print(a)