[python 파이썬] 조건문 if문 한 줄로 정리

파이썬에서 조건문은 if 문과 elif 문 혹은 else 문으로 이루어지며 보통은 각각을 한 줄씩 적는다.

예를 들면 아래와 같다.

 

Score = 100

if Score == 100:
    Grade = 'A'
elif Score > 70:
    Grade = 'B'
else:
    Grade = 'C'
    
print(Grade) ##'A'

 

하지만 간단한 조건문을 위해 6줄이나 소요해야한다.

이것을 조금 더 간략화하기 위해 콜론 옆에 Grade를 정의해줄 수 있다.

 

Score = 100

if Score == 100: Grade = 'A'
elif Score > 70: Grade = 'B'
else           : Grade = 'C'
    
print(Grade)##'A'

 

첫 번째 코드보다 두 번째 코드가 더 간략한 걸 확인할 수 있다. 그렇다면 if-elif-else 구문을 한 줄로도 표현할 수 있을까?

 

 

그렇다. 아래와 같은 공식으로 한 줄 표현이 가능하다.

Result  =  A  if  조건1  else  B  if  조건2  else  C

이는 조건 1이 True이면 Result는 A이고 조건1이 False이고 조건 2가 True이면 Result는 B이고 조건 1, 2 모두 False이면 Result가 C라는 뜻이다. 위 공식을 사용하여 예제를 구현해보면 아래와 같다.

 

Score = 100

Grade = 'A' if Score == 100 else 'B' if Score > 70 else 'C'
    
print(Grade)##'A'

 

만약 if-else문만 한줄로 표현하고 싶다면 Result = A if 조건 else B로 표현할 수 있다:)