파이썬 조건문(if), 반복문(for) 줄이기

파이썬 조건문(if), 반복문(for) 줄이기

 

조건문과 반복문을 줄여서 쓰는 경우를 많이 보진 못했다. 그래도 생각보다 직관적이고, 연습만 된다면 더 깔끔한 코드를 짜는데 도움이 될 것 같다.

 

# if문 삼항연산자

num = 3

if num % 2 == 0:
    result = '짝수'
else:
    result = '홀수'
    
print(f'{num}은 {result}입니다.')

######### 아래는 삼항연산자 #########

result = ('짝수' if num % 2 == 0 else '홀수')   # 직관적이다. 괄호가 없어도 된다.

print(f'{num}은 {result}입니다.')


#######################################################################


# for문 줄이기

a_list = [1, 3, 2, 5 ,1, 2]

# 일반적인 for문 형태
b_list = []
for a in a_list:
    b_list.append(a*2)

# 줄인 형태    
b_list = [a*2 for a in a_list] # a_list에 있는 요소들 a를 *2해서 리스트안에 넣는다.

print(b_list)

 

개인적으로 조건문보다 반복문이 더 직관적이다. 일단 일반적인 형태로 작성하고 줄여보는 식으로 연습을 해봐야겠다.