본문 바로가기

Programming/Python

입력값이 몇 개가 될지 모를 때(*args)

여러 개의 입력값을 받는 함수

def add_many(*args):

	result = 0
    
    for i in args:
        result = result + i
        
    return result

입력 값이 몇 개이든 상관이 없음

*args처럼 * 뒤로 아무 이름이나 붙여 쓰면 입력값을 전부 모아서 튜플로 만들어줌

 

 

응용하기

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
    
    
print(add_mul('add', 1, 2, 3, 4, 5))
print(add_mul('mul', 1, 2, 3, 4, 5))

'Programming > Python' 카테고리의 다른 글

os모듈 함수  (0) 2020.09.18
튜플  (0) 2020.09.15
영화평점 분석하기  (0) 2020.03.09
Pandas  (0) 2020.03.09
Numpy  (0) 2020.03.09