[기본이론] defaultdict()의 활용법

참고 : dongdongfather.tistory.com/69

[파이썬 기초] 유사 딕셔너리 defaultdict() 활용법

해당 내용을 바탕으로 정리하였다.

 

코딩테스트 문제를 풀다가, 때로는 리스트에 담아서 key를 부여하여 dict로 묶어서 한번에 정리하는 것이 편한 경우가 있고, (특히 그래프 문제에서 이것이 기본 세팅이다) 이를 위한 방법이 defaultdict이다.

 

defaultdict는 collections에 있는 모듈이므로, import해서 불러와야 한다.

defaultdict의 가장 큰 특징은, 키를 지정할때 값을 주지 않을 경우, 디폴트 값을 지정할 수 있다는 것이다.

예시)

list_dict = defaultdict(list)
list_dict['key1']

이 경우 key1에 아무것도 지정하지 않았으므로, 'key1'에는 빈 리스트가 들어간다!

 

예시)

list = [(1, 'a'), (2, 'c'), (1, 'b')]
l_dict = defaultdict(list)

for k, v in list:
    l_dict[k].append(v)

이 경우 l_dict는 { 1: ['a', 'b'], 2: ['c'] } 이런식으로 리스트 안에 담기는 형태로 저장된다. list가 기본형이기에  이런식으로 구분해서 담을 수 있다.