if문 한 줄로 쓰기 (one-liner)

one-liner

한 변수에 값을 할당할 때

# traditional if-else
if x > 0:
    value = 10
else:
    value = 20

# one-liner
value = 10 if x > 0 else 20

if-elif-else

# traditional if-elif-else
alpha = 10

if alpha > 7:
    beta = 999
elif alpha == 7:
    beta = 99
else:
    beta = 0


# one-liner
beta = 999 if alpha > 7 else (beta == 99 if alpha == 7 else 0)

dictionary를 이용한 if-elif-else

# traditional if-elif-else
alpha = 7

if alpha > 7:
        beta = 999
elif alpha == 7:
        beta = 99
else:
        beta = 0


# one-liner
beta = {alpha > 7: 999, alpha < 7: 0}.get(True, 99)

참고 : https://medium.com/@epicshane/one-liner-for-python-if-elif-else-statements-d9d46016ba2a