Unit 30. 함수에서 위치 인수와 키워드 인수 사용하기

지금까지 간단하게 'Hello, world!'를 출력하는 함수와 두 수를 더하는 함수를 만들어보았습니다. 파이썬에서는 함수를 좀 더 편리하게 사용할 수 있도록 다양한 기능을 제공합니다. 이번에는 함수에서 위치 인수, 키워드 인수를 사용하는 방법과 리스트, 딕셔너리 언패킹(unpacking)을 활용하는 방법을 알아보겠습니다.

30.1 위치 인수와 리스트 언패킹 사용하기

다음과 같이 함수에 인수를 순서대로 넣는 방식을 위치 인수(positional argument)라고 합니다. 즉, 인수의 위치가 정해져 있습니다.

>>> print(10, 20, 30)
10 20 30

print에 10, 20, 30 순으로 넣었으므로 출력될 때도 10 20 30으로 출력됩니다. 우리가 지금까지 흔히 사용한 방식이죠.

30.1.1  위치 인수를 사용하는 함수를 만들고 호출하기

그럼 간단하게 숫자 세 개를 각 줄에 출력하는 함수를 만들어보겠습니다.

>>> def print_numbers(a, b, c):
...     print(a)
...     print(b)
...     print(c)
...

print_numbers에 숫자 세 개를 넣으면 각 줄에 숫자가 출력됩니다.

>>> print_numbers(10, 20, 30)
10
20
30

30.1.2  언패킹 사용하기

이렇게 인수를 순서대로 넣을 때는 리스트나 튜플을 사용할 수도 있습니다. 다음과 같이 리스트 또는 튜플 앞에 *(애스터리스크)를 붙여서 함수에 넣어주면 됩니다.

• 함수(*리스트)
• 함수(*튜플)

>>> x = [10, 20, 30]
>>> print_numbers(*x)
10
20
30

print_numbers에 10, 20, 30이 들어있는 리스트 x를 넣고 *만 붙였는데도 숫자가 각 줄에 출력되었습니다. 즉, 리스트(튜플) 앞에 *를 붙이면 언패킹(unpacking)이 되어서 print_numbers(10, 20, 30)과 똑같은 동작이 됩니다. 말 그대로 리스트의 포장을 푼다는 뜻입니다.

리스트 변수 대신 리스트 앞에 바로 *를 붙여도 동작은 같습니다.

>>> print_numbers(*[10, 20, 30])
10
20
30

단, 이때 함수의 매개변수 개수와 리스트의 요소 개수는 같아야 합니다. 만약 개수가 다르면 함수를 호출할 수 없습니다. 여기서는 함수를 def print_numbers(a, b, c):로 만들었으므로 리스트에는 요소를 3개 넣어야 합니다. 다음과 같이 요소가 두 개인 리스트를 넣으면 에러가 발생합니다.

>>> print_numbers(*[10, 20])
Traceback (most recent call last):
  File "<pyshell#16>", line 1, in <module>
    print_numbers(*[10, 20])
TypeError: print_numbers() missing 1 required positional argument: 'c'

30.1.3  가변 인수 함수 만들기

그럼 위치 인수와 리스트 언패킹은 어디에 사용할까요? 이 기능들은 인수의 개수가 정해지지 않은 가변 인수(variable argument)에 사용합니다. 즉, 같은 함수에 인수 한 개를 넣을 수도 있고, 열 개를 넣을 수도 있습니다. 또는, 인수를 넣지 않을 수도 있습니다.

다음과 같이 가변 인수 함수는 매개변수 앞에 *를 붙여서 만듭니다.

def 함수이름(*매개변수):
    코드

이제 숫자 여러 개를 받고, 숫자를 각 줄에 출력하는 함수를 만들어보겠습니다. 다음과 같이 함수를 만들 때 괄호 안에 *args와 같이 매개변수 앞에 *를 붙입니다. 그리고 함수 안에서는 for로 args를 반복하면서 print로 값을 출력합니다.

>>> def print_numbers(*args):
...     for arg in args:
...         print(arg)
...

매개변수 이름은 원하는 대로 지어도 되지만 관례적으로 arguments를 줄여서 args로 사용합니다. 특히 이 args는 튜플이라서 for로 반복할 수 있습니다.

그럼 print_numbers 함수에 숫자를 넣어서 호출해봅니다. 숫자를 한 개 넣으면 한 개 출력되고, 네 개 넣으면 네 개가 출력됩니다. 즉, 넣은 숫자 개수만큼 출력됩니다.

>>> print_numbers(10)
10
>>> print_numbers(10, 20, 30, 40)
10
20
30
40

이렇게 함수에 인수 여러 개를 직접 넣어도 되고, 리스트(튜플) 언패킹을 사용해도 됩니다. 다음과 같이 숫자가 들어있는 리스트를 만들고 앞에 *를 붙여서 넣어봅니다.

>>> x = [10]
>>> print_numbers(*x)
10
>>> y = [10, 20, 30, 40]
>>> print_numbers(*y)
10
20
30
40

리스트에 들어있는 값이 그대로 출력되었습니다. 즉, 리스트 x는 [10]이므로 print_numbers(*x)로 호출하면 print_numbers(10)과 같고 리스트 y는 [10, 20, 30, 40]이므로 print_numbers(10, 20, 30, 40)과 같습니다.

이처럼 함수를 만들 때 def print_numbers(*args):와 같이 매개변수에 *를 붙여주면 가변 인수 함수를 만들 수 있습니다. 그리고 이런 함수를 호출할 때는 인수를 각각 넣거나, 리스트(튜플) 언패킹을 사용하면 됩니다.

>>> def print_numbers(a, *args):
...     print(a)
...     print(args)
...
>>> print_numbers(1)
1
()
>>> print_numbers(1, 10, 20)
1
(10, 20)
>>> print_numbers(*[10, 20, 30])
10
(20, 30)