Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

파이썬으로 리스트 평균 구하기 – sum() 함수 활용 방법

파이썬은 여러 개의 숫자 요소를 손쉽게 더할 수 있도록 sum() 함수를 기본적으로 제공합니다. 이 함수를 활용하면 리스트에 담긴 모든 요소의 합계를 구한 뒤, 요소 개수로 나누어 간단하게 평균값을 계산할 수 있습니다.

알고리즘

1단계: 리스트의 크기(size)를 입력받는다.
2단계: 리스트에 들어갈 숫자(요소)를 입력받는다.
3단계: sum() 함수를 사용해 모든 숫자의 합계를 계산한다.
4단계: 합계를 개수로 나누어 평균을 구한다.

예제 코드

# 리스트의 평균 구하기
A = list()
n = int(input("리스트의 크기를 입력하세요 :: "))
print("숫자를 입력하세요 ::")
for i in range(int(n)):
    k = int(input(""))
    A.append(int(k))
sm = sum(A)
avg = sm / n
print("SUM = ", sm)
print("AVERAGE = ", avg)

실행 결과

리스트의 크기를 입력하세요 :: 5
숫자를 입력하세요::
10
20
30
40
50
SUM = 150
AVERAGE = 30.0

코드 설명

위 코드는 다음과 같은 순서로 동작합니다.

먼저 빈 리스트 A를 생성하고, 사용자로부터 리스트의 크기 n을 입력받습니다. 이후 for 반복문을 통해 n번만큼 숫자를 입력받아 append() 메서드로 리스트에 추가합니다.

모든 요소가 저장되면 sum(A)로 전체 합계를 구하고, 합계를 n으로 나누어 평균을 계산합니다. 마지막으로 합계와 평균을 화면에 출력합니다.

참고: statistics 모듈 활용하기

파이썬 3.4 이상에서는 표준 라이브러리인 statistics 모듈의 mean() 함수를 사용해 더욱 간결하게 평균을 구할 수도 있습니다.

import statistics
avg = statistics.mean([10, 20, 30, 40, 50])
print(avg)  # 30 출력

두 방법 모두 결과는 동일하지만, statistics.mean()을 사용하면 코드가 더 짧아지고 의미도 명확해진다는 장점이 있습니다.