[파이썬] 문제 : 가변인자 * 사용해서 과일 출력

(문제)

문제: 가변인자*를 사용하여 최소 3가지 이상의 과일을 입력받아 입력받은 과일을 출력하시오. 만약 입력받은 과일중 "apple"이 있다면 "apple"출력후 "RED"를 출력하시오.

 

 

(방법)

def fruit(*name):
    for i in range(len(name)):     # 가변인자는 리스트 처럼 생각하면 됩니다
        if name[i] == 'apple':     # 인수로 받은 가변인자가 apple 면
            print(name[i], 'RED')   # RED 를 붙임
        else:
            print(name[i])
    
fruit('banana','apple','watermelon')