[COS Pro 2급, Python] 1차 10번 : 평균 이하의 개수 구하기

문제

평균은 자료의 합을 자료의 개수로 나눈 값을 의미합니다.

자연수가 들어있는 배열의 평균을 구하고, 평균 이하인 숫자는 몇개 있는지 구하려합니다.

예를 들어 주어진 배열이 [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]이라면, 평균은 5.5이므로 배열에서 평균 이하인 값은 5개입니다.

자연수가 들어있는 배열 data와 data의 길이 data_len이 매개변수로 주어질 때, 배열에 평균 이하인 값은 몇 개인지 return 하도록 solution 함수를 작성했습니다. 그러나, 코드 일부분이 잘못되어있기 때문에, 몇몇 입력에 대해서는 올바르게 동작하지 않습니다.

주어진 코드에서 세줄만 변경해서 모든 입력에 대해 올바르게 동작하도록 수정하세요.

 

 

code

def solution(data):
	total = sum(data)
	average = total / len(data)
	cnt = 0
	for d in data:
		if d <= average:
			cnt += 1
	return cnt


data1 = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
ret1 = solution(data1)

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

data2 = [1, 1, 1, 1, 1, 1, 1, 1, 1, 10]
ret2 = solution(data2)

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