[Python] 함수에 리스트 입력받기

python에서 함수를 사용할 때 튜플 형태로 밖에 못 불러온다

def add_all(*args):
	num = len(args)
    sum = 0
    for i in range(0, num):
    	sum += args[i]
    return sum

이러한 함수로 튜플의 요소를 모두 합하는 함수를 선언할 수 있다.

add_all(1, 2, 3, 4, 5, 6, 7)

28

튜플 뿐만 아니라 리스트로 요소를 받을 수도 있다.

def add_all_list(*lst):
	sum = 0
    for i in range(len(lst)):
    	for j in lst[i]:
        	sum += j
    return sum

이 코드로 배열을 이용하면 리스트도 입력 받을 수 있다.

add_all_list([1, 2, 3, 4, 5])

15

리스트, 튜플 상관없이 함수로 입력받기

def add_all_both(*args):
	temp = 0
    for i in range(len(args)):
    	if type(args[i]) ==list:
        	for j in args[i]:
            	temp+=j
        else:
        	temp += args[i]
   return temp

이 코드를 사용하면 입력하는 항목의 타입이 리스트여도 튜플이여도 계산이 가능해진다