[프로그래머스] #12 평균 구하기

▶ 문제

정수를 담고 있는 배열 arr의 평균값을 return하는 함수 solution 만들기

 

▶ 제한사항

arr은 길이 1 이상, 100 이하인 배열

arr의 원소는 -10,000 이상 10,000 이하인 정수

 

▶ 풀이

def solution(arr):
    return sum(arr)/len(arr)

 

sum(arr) : 배열 arr 내 요소들의 합을 구해주는 함수

len(arr) : 배열 arr의 길이(요소들의 갯수)를 구해주는 함수

 

제한사항에 배열 arr의 길이가 1 이상이라는 말이 있어서 고려 안했는데,

만약 배열 arr의 길이가 0일 경우에는 Zero division Error 가 생겨버린다.

def solution(arr):
    if len(arr) == 0:           # 분모가 0일경우
        return 0
    return sum(arr)/len(arr)

 

그럴때는 이렇게 길이가 0일 경우를 따로 조건을 걸어준다.

 

 

 

 

---