평균 구하기 / 프로그래머스, 파이썬 코딩테스트 연습 연습문제

문제

정수를 담고 있는 배열 arr의 평균값을 return하는 함수, solution을 완성해보세요.

 

 

제출답안

def solution(arr):
    # 정수인 숫자들의 리스트
    ls_valid = [num for num in arr if num % 1 == 0.0]
    
    # 평균값 계산
    answer = sum(ls_valid) / len(ls_valid)
    return answer

 

 

제출 후 개선답안

작성한 코드는 정수가 아닌 것을 구별하기 위해 if 문을 추가하였음
하지만 문제에서는 배열 내 정수만 주어진다 하였으므로
정수 여부를 판별하는 if 문을 제거하여 코드를 간소화하였음

def solution(arr):
    # 정수인 숫자들의 리스트
    ls_valid = [num for num in arr]
    
    # 평균값 계산
    answer = sum(ls_valid) / len(ls_valid)
    return answer

 

 

 

문제 바로가기