[COS PRO Python 기출 - 1차] 1. 단체 티셔츠를 주문하기

goorm

 

---

---

코드

def solution(shirt_size):
	answer = [0]*6            # [0, 0, 0, 0, 0, 0]으로 만들기
	size_list = ['XS', 'S', 'M','L','XL','XXL']
	for i in range(len(shirt_size)):
		for j in range(len(size_list)):
			if shirt_size[i] == size_list[j]:
				answer[j] += 1
	return answer
    
shirt_size = ["XS", "S", "L", "L", "XL", "S"]
ret = solution(shirt_size);

print("solution 함수의 반환 값은", ret, "입니다.")

 

풀이

• answer를 [0, 0, 0, 0, 0, 0]로 만들어서, 각 티셔츠 사이즈 별로 갯수를 answer 리스트에 추가할 수 있도록 함
• 이중 for문을 활용하여 shirt_size의 각 원소가 size_list에 해당하는 사이즈가 있는지, 있다면 그 사이즈의 인덱스에 갯수를 추가함