[Python] BOJ / 2108번 / 통계학

2108번: 통계학

 

📝 문제

수를 처리하는 것은 통계학에서 상당히 중요한 일이다. 통계학에서 N개의 수를 대표하는 기본 통계값에는 다음과 같은 것들이 있다. 단, N은 홀수라고 가정하자.

1. 산술평균 : N개의 수들의 합을 N으로 나눈 값
2. 중앙값 : N개의 수들을 증가하는 순서로 나열했을 경우 그 중앙에 위치하는 값
3. 최빈값 : N개의 수들 중 가장 많이 나타나는 값
4. 범위 : N개의 수들 중 최댓값과 최솟값의 차이 

N개의 수가 주어졌을 때, 네 가지 기본 통계값을 구하는 프로그램을 작성하시오.

 

📜 풀이

• 문제에 나와있는대로 산술평균, 중앙값, 최빈값, 범위를 계산하여 출력해주었다.
• 최빈값의 경우에는 파이썬에서 제공하는 Counter를 사용하였다.

 

💻 소스코드

from collections import Counter
import sys
input = sys.stdin.readline

def get_mode():
    counter = Counter(nums)
    elements = counter.most_common()

    modes = []
    for num, cnt in elements:
        if cnt == elements[0][1]:
            modes.append(num)
    
    return modes[0] if len(modes) == 1 else modes[1]

n = int(input())
nums = [int(input()) for _ in range(n)]

# 산술평균
print(round(sum(nums) / n))

# 중앙값
nums.sort()
print(nums[n // 2])

# 최빈값
print(get_mode())

# 범위
print(nums[-1] - nums[0])