백준(baekjoon) 2108번: 통계학(실버 3, 파이썬 PYTHON) - 정렬

문제

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

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

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

 

https://www.acmicpc.net/problem/2108

2108번: 통계학

 

import sys

N=int(sys.stdin.readline())
num_list=[]
sum=0

for _ in range(N):
  num=int(sys.stdin.readline())
  num_list.append(num)

for x in num_list:
  sum+=x 

print(round(sum/N)) 

num_list.sort() 
print(num_list[N//2]) 

dict={}
for x in num_list:
  if x in dict:  
    dict[x]+=1  
  else: 
    dict[x]=0 

array_values=list(dict.values()) 
array_values.sort() 
array_keys=[]

for x in dict:
  if(dict[x]==array_values[len(array_values)-1]): 
    array_keys.append(x)  

array_keys.sort() 
if(len(array_keys)==1):  
  print(array_keys[0])  
else:  
  print(array_keys[1])  

print(num_list[N-1]-num_list[0])

 

※ 정렬

 

 

● 먼저 입력받은 숫자들을 num_list라는 배열에 저장해준다.

 

★ 산술평균

 

● sum에 num_list의 숫자를 모두 더해준 후, N으로 나눈 값을 round() 함수를 이용해 반올림 하고 출력

 

★ 중앙값

 

● 먼저 num_list를 sort() 함수를 이용하여 정렬한다. N은 홀수이므로 N//2를 인덱스로 하는 num_list의 요소는 중앙값이 된다.

 

★ 최빈값

 

● 우선 dict라는 파이썬 딕셔너리를 하나 정의해준다.

 

● 그 후 num_list를 순회하며 숫자가 dict의 key에 존재하면 그 key의 value값을 1 증가시키고 만약 존재하지 않으면 그 숫자를 key로 하는 value가 0인 요소를 dict에 추가해준다. (value는 그 key 값이 몇 개나 입력되었는지 세는 용도.)

 

● 그 후, array_values라는 배열에 dict의 value만 따로 담고, 정렬시켜준다.

 

● 그리고 dict의 key들을 돌며 그 key의 value 값이 array_values 배열의 끝 요소의 값과 같으면

array_keys 라는 배열에 그 key값을 append 해준다.

 

● 마지막으로 array_keys를 sort() 함수를 이용해 정렬해준 뒤, 만약 array_keys의 길이가 1이면 최빈값이 하나인 것이므로 그 안에 있는 요소를 출력, 만약 길이가 1 이상이면 최빈값이 여러개란 뜻이므로 그 중 두번째로 작은 값인 array_keys[1]을 출력해준다.

 

★ 범위

 

● num_list는 정렬되어 있으므로 맨 끝 요소와 맨 처음 요소를 빼면 범위가 나온다.

 

 

결과

 

 

♣ 참고: 파이썬 딕셔너리

https://wikidocs.net/16

02-5 딕셔너리 자료형