31.4 함수에서 값을 여러 개 반환하기

지금까지 함수의 결과를 변수에 저장할 때 값을 한 개만 저장했습니다. 이번에는 함수에서 값을 여러 개 반환하는 방법을 알아보겠습니다. 함수에서 값을 여러 개 반환할 때는 다음과 같이 return에 값이나 변수를 ,(콤마)로 구분해서 지정하면 됩니다.

def 함수이름(매개변수):
    return 반환값1, 반환값2

그럼 두 수를 더한 값과 뺀 값을 반환하는 함수를 만들어보겠습니다. 다음과 같이 return으로 ab를 더한 값과 a에서 b를 뺀 값을 반환하도록 만듭니다.

>>> def add_sub(a, b):
...     return a + b, a - b
...

이제 add_sub 함수에 10과 20을 넣고 결과를 xy에 저장합니다. 그리고 xy의 값을 출력해봅니다.

>>> x, y = add_sub(10, 20)
>>> x
30
>>> y
-10

x의 값을 출력해보면 10과 20을 더한 값인 30이 나오고, y의 값을 출력해보면 10에서 20을 뺀 값인 -10이 나옵니다. 이처럼 return은 값을 여러 개 반환할 수 있습니다.

add_sub 함수의 호출 과정을 그림으로 표현하면 다음과 같은 모양이 됩니다.

그림 31-4 반환값 여러 개를 변수 여러 개에 저장
그림 31 4 반환값 여러 개를 변수 여러 개에 저장

이렇게 return으로 값을 여러 개 반환하면 실제로는 튜플로 반환됩니다. 다음과 같이 add_sub의 결과를 변수 한 개에 저장해서 출력해보면 튜플이 반환되는 것을 알 수 있습니다.

>>> x = add_sub(10, 20)
>>> x
(30, -10)

즉, 튜플이 변수 여러 개에 할당되는 특성을 이용한 것입니다(언패킹).

>>> x, y = (30, -10)
>>> x
30
>>> y
-10

물론 함수의 return에서 리스트(튜플)을 직접 반환한 뒤 변수 여러 개에 값을 저장해도 됩니다.

>>> def one_two():
...     return [1, 2]
...
>>> x, y = one_two()
>>> print(x, y)
1 2

참고로 파이썬에서 값을 콤마로 구분하면 튜플이 됩니다. 따라서 return 1, 2return (1, 2)처럼 튜플을 반환하는 것과 같은 의미입니다.

>>> 1, 2
(1, 2)