[프로그래머스/Programmers] 코딩테스트 입문 > 최빈값 구하기 (파이썬 Python)

문제

https://school.programmers.co.kr/learn/courses/30/lessons/120812

프로그래머스

 

 

내 제출

- 배열에서 각 숫자의 개수를 쉽게 계산하기 위해 Counter 사용.

from collections import Counter

def solution(array):
    count = Counter(array)
    max_count = max(count.values())

    modes = [num for num, freq in count.items() if freq == max_count]

    if len(modes) > 1:
        return -1
    else:
        return modes[0]

 

 

코드 설명

from collections import Counter

- 'collections' 모듈에서 'Counter' 클래스 가져와 사용.

- 'Counter'클래스 : 주어진 시퀀스(리스트, 튜플 등)나 반복 가능한 객체에서 각 원소의 개수를 세는 기능 제공.

 

count = Counter(array)
max_count = max(count.values())

- count : 배열 array의 각 숫자의 개수를 세고, 그 결과를 count 변수에 저장

- count.values() : count 딕셔너리의 값(각 숫자의 개수)들을 가져옴.

- max_count : max() 함수를 사용하여 가장 큰 값을 찾음 => 최빈값의 개수

 

modes = [num for num, freq in count.items() if freq == max_count]

- count.items() : count 딕셔너리의 각 항목을 가져옴 ← 각 항목은 (숫자, 개수)의 형태의 튜플

- 리스트 컴프리헨션을 사용(https://taetaegom.tistory.com/25) → 개수가 최빈값과 같은 숫자들만 modes 에 넣음.

- count.item()의 num, freq에서 num을 list에 넣음, 단 조건은 freq==mqx_count.

 

 

참고

https://www.daleseo.com/python-collections-counter/

파이썬 collections 모듈의 Counter 사용법