[Python] 함수 - 입력값이 몇 개가 될지 모를 때

파이썬은 이런 문제를 해결하기 위해 다음과 같은 방법을 제공한다.

def 함수이름(*매개변수):
   수행할 문장
   ...

 

 

-여러 개의 입력값을 받는 함수 만들기

>>> def add_many(*args):
...     result = 0
...     for i in args:
...         result = result + i  # *args에 입력받은 모든 값을 더한다
...     return result
...
>>>

*여기서 *args는 임의로 정한 변수 이름이다.

*args처럼 매개변수 이름 앞에 *을 붙이면 입력값을 전부 모아서 튜플로 만들어 준다.

 

<함수를 적용한 것>

>>> result = add_many(1,2,3)  # add_many 함수의 결괏값을 result 변수에 대입
>>> print(result)
6
>>> result = add_many(1,2,3,4,5,6,7,8,9,10)
>>> print(result)
55

 

 

여러 개의 입력을 처리할 때 def add_many(*args)처럼 함수의 매개변수로 *args만 사용할 수 있는 것은 아니다.

>>> def add_mul(choice, *args):
...     if choice == "add":  # 매개변수 choicedp 'add'를 입력받았을 때
...         result = 0
...         for i in args:
...             result = result + i  # args에 입력받은 모든 값을 더한다
...     elif choice == "mul":  # 매개변수 choicedp 'mul'을 입력받았을 때
...         result = 1
...         for i in args:
...             result = result * i  # *args에 입력받은 모든 값을 곱한다
...     return result
...
>>>

>>> result = add_mul('add', 1,2,3,4,5)
>>> print(result)
15
>>> result = add_mul('mul', 1,2,3,4,5)
>>> print(result)
120

매개변수 choice에 'add'가 입력된 경우 *args에 입력되는 모든 값을 더해서 15를 돌려주고,

'mul'이 입력된 경우 *args에 입력되는 모든 값을 곱해서 120을 돌려준다.