Python :: 함수 / *args

점프투파이썬 참고하여 복습

 

 

 

입력값이 몇 개가 될 지 모를 때는 여러개의 입력값을 받는 함수 만들기

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

 

from ast import arg


def add_many(*args):
    result = 0
    for i in args:
        result += i
    print(result)
    
add_many(1,10,20)
add_many(1,10)

31

11

 

 

 

 

 

*args 매개변수 앞에 choice 매개변수 추가

from unittest import result


def add_mul(choice, *args):
    if choice == "add": #매개변수 choice에 'add'를 입력 받을때
        result = 0
        for i in args:
            result += i
    elif choice == "mul": #매개변수 choice에 'mul'을 입력 받을때
        result = 1
        for i in args:
            result *= i
            
    return result

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

15
120

 

 

---

return의 또 다른 쓰임새

def say_nick(nick):
    if nick == "바보":
        return
    print("나의 별명은 %s 입니다." %nick)
    
say_nick("야호")
say_nick("호호")
say_nick("바보")
say_nick("안녕")

나의 별명은 야호 입니다.
나의 별명은 호호 입니다.
나의 별명은 안녕 입니다.

 

 

if에 해당할 때 return => 함수를 빠져나간다