본문 바로가기
Language/Python

2. 파이썬 딕셔너리 모듈 defaultdict, Counter

by 규나 2021. 7. 22.
SMALL

defaultdict

존재하지 않는 키를 조회할 경우, 에러 메시지 대신 디폴트 값을 기준으로 해당 키에 대한 아이템을 생성해준다.
아래의 예제 코드에서 C는 존재하지 않는 키 이지만 디폴트인 0을 기준으로 자동 생성한 후 1을 더해 아이템이 만들어진다.

a = collections.defaultdict(int)
a['A'] = 5 
a['B'] =4
>>> a
defaultdict{<class 'int'>, {'A':5, 'B':4})

>>> a['C'] += 1
>>> a
defaultdict{<class 'int'>, {'A':5, 'B':4, 'C':1})

 

Counter

아이템에 대한 개수를 계산해 딕셔너리로 리턴해준다.
most_common(n) 메서드를 사용하면, 가장 빈도가 높은 n개의 요소를 리턴한다.

a = [1,2,3,4,5,5,5,6,6]
b = collections.Counter(a)
>>> b
Counter({5:3, 6:2, 1:1, 2:1, 3:1, 4:1})

>>> b.most_common(2)
[(5,3), (6,2)]

 

댓글