Python/프로그래머스

[programmers] LV 1. 평균 구하기 (Python/파이썬)

이니니 2022. 3. 8. 18:53
문제

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

 

제한 사항

· arr은 길이 1 이상, 100 이하인 배열입니다.
· arr의 원소는 -10,000 이상 10,000 이하인 정수입니다.

 

입출력 예

arr return
[1, 2, 3, 4] 2.5
[5, 5] 5

 

▶ my code

def solution(arr):
    answer = 0
    for i in arr:
        answer += i
    answer /= len(arr)
    return answer

 

▶ 아이디어

각각의 배열의 원소를 더하고, 그것을 배열의 길이로 나누어 주었다.