[Programmers/프로그래머스]평균 구하기

문제 설명

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

제한사항

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

입출력 예

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

풀이

이 문제는 arr의 요소들의 평균을 구하여 반환하는 문제이다. 위 문제는 2가지 풀이 방법이 있다.

첫번째는 sum()과 len() 함수를 이용하여 구할 수 있다. sum()으로 요소의 합을 구한 후 len()으로 전체 길이를 구하여 나눠주면 평균을 구할 수 있다.

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

두번째는 numpy 라이브러리를 쓰는 방법이다. numpy의 array() 메소드로 arr를 numpy array로 만든 후 mean() 메소드로 평균을 구하면 된다.

import numpy as np
def solution(arr):
	return np.array(arr).mean()