Python - 함수의 파라미터 여러개 일때(*args)

보통 함수라면 인수(파라미터)가 몇개인지 정해져있다.

더하기 함수를 만든다고 가정하면

def add_cal(a, b):
    return a + b

a하고 b로 들어가는 수가 인수이다. 들어갈 수 있는 파라미터는 2개이다.

그러면 인수가 몇개가 들어갈 지 모를때는?

 

def add_cal(*args):
    result = 0
 
    for i in args:
        result = result + i
    return result
    

 

예를 들어,

인수의 개수를 그냥 마음대로 1, 2, 3, 4, 5 라는 5개를 넣어보고 이걸 싹다 더해서 result로 반환

그리고 출력해보았다. 15가 나온다.

def add_cal(*args):
    result = 0
 
    for i in args:
        result = result + i
    return result
 
print(add_cal(1, 2, 3, 4, 5))
 
 
>>>15

 

-------------------------------------------------------------------------------------------------------------------

이것도 가능하다!

def add_mul(choice, *args):
    if choice == "add":
        result = 0
        for i in args:
            result = result + i
 
    elif choice == "mul":
        result = 1
        for i in args:
            result = result * i
 
    return result

 

add_mul('mul', 1, 2, 3, 4, 5)