defaultdict 정리

코딩테스트 관련 문제를 공부하다가 다른분의 코드를 보게되었고 defaultdict를 사용하는 것을 보고 궁금하여 공부를 하게 되었다.

python의 collections 모듈에는 defaultdict라는 기능이 있다.

defaultdict는 이름 그대로 기본값을 지정한 딕셔너리인데 기본 딕셔너리와 어떠한 차이가 있는지 알아보자.

defaultdict()는 딕셔너리를 만드는 dict클래스의 서브클래스이다.

작동하는 방식은 기본 딕셔너리와 거의 동일한데 defaultdict()는 인자로 주어진 객체(default-factory)의 기본값을 딕셔너리값의 초깃값으로 지정할 수 있다.

-> 이게 무슨말이냐면 일반적인 딕셔너리의 경우 미리 삽입하지 않은 key를 호출하면 다음과 같이 에러가 발생한다.

>>> n_dict = dict()
>>> n_dict["a"]

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
KeyError: 'a'

“a” 라는 키의 값을 넣어준적이 없기때문에, 당연히 에러가 발생하는 것이다.

이에 setdefault를 사용해 호출과 동시에 선언도 가능하다.

>>> n_dict = dict()
>>> n_dict.setdefault("a", 0)
0
>>> n_dict
{'a': 0}

하지만 이 방법을 계속 사용하는 것은 그다지 효율적이지 않다.

이때 defaultdict를 활용하는 것이다.

defaultdict를 활용해 다음과 같이 기본값을 ‘int’ 로 선언해주고, 기존에 없던 key를 호출하면 다음과 같이 해당 key가 0으로 자동 초기화된다.

-> 기본값을 int뿐만 아니라 리스트, set(집합) 등으로 초기화 할 수 있기 때문에 여러 용도로 사용할 수 있다.

>>> from collections import defaultdict
>>> d_dict = defaultdict(int)
>>> d_dict["a"]
0
>>> d_dict
defaultdict(<class 'int'>, {'a': 0})

default값으로 list를 주었을때 >>> list_dict = defaultdict(list) - 값을 지정하지 않으면 []로 초기화

default값으로 set을 주었을때 >>> list_dict = defaultdict(set) - 값을 지정하지 않으면 set()으로 초기화

defaultdict는 외부함수이기 때문에 사용하기 위해서는 import과정이 필요하다.

>>> from collections import defaultdict

이뿐만 아니라 미리 선언하지 않은 key에 값을 더해주는 것도 가능하다.

>>> d_dict = defaultdict(int)
>>> d_dict["a"] += 10
10
>>> d_dict
defaultdict(<class 'int'>, {'a': 10})

lambda 식을 사용해 원하는 초기값을 지정할수도 있다.

>>> d_dict = defaultdict(lambda: 'default value')
>>> d_dict["a"]
'default value'

해당 내용은 위의 사이트를 참고하였다.

HyunWook.dev