[ 파이썬(Python) ] 함수(def)에서 return 인자의 수가 다를 때

📍 함수(def)에서 return 인자의 수가 다를 때

코딩 테스트 문제를 풀다 보면 False일때 -1만 return 시키고 True면 a, b처럼 1개 이상을 return해야 할 때 다음과 같이 작성할 수 있다.

만약, a+b값이 20을 넘으면 a, b을 리턴해야하고, 아니면 -1을 리턴해야한다면 다음과 같은 코드가 될 것이다.

a, b = map(int, input().split())

def check():
    if a + b >= 20:
        return a, b
    else:
        return -1

print(check())

 

하지만, -1의 리턴값에 임의의 수 0을 넣고 똑같이 2개를 리턴하게 작성한 다음 첫 번째 값이 -1이면 -1을 출력시키고 아니면 c, d를 출력하게 할 수 있다.

728x90

a, b = map(int, input().split())

def check():
    if a + b >= 20:
        return a, b
    else:
        return -1, 0

c, d = check()
if c == -1:
    print(-1)
else:
    print(c, d)