[Python] 자료구조 - collections 모듈 - defaultdict

[defaultdict]

딕셔너리 구조를 사용할 때에는 키를 먼저 생성한 다음, 생성된 키를 호출할 수 있습니다. 만약 키를 생성하지 않았는데도 키를 호출하려고 하면 에러가 발생합니다. 그러나 defaultdict는 기본 값을 설정할 수 있습니다. 따라서 생성한 적이 없는 키를 호출하게 되면, defaultdict 선언 시에 지정한 기본값이 반환됩니다.

from collections import defaultdict

d = defaultdict(lambda: 0)
print(d['a'])
'''
>>> 0
'''

 

defaultdict에서는 위의 코드처럼 '0'이라는 구체적인 숫자를 지정하지 않고, 특정한 자료형을 기본값으로 설정할 수도 있습니다. 만약 int를 기본값으로 설정하게 되면, 생성한 적 없는 키를 호출했을 때 0을 반환합니다.

from collections import defaultdict

d = defaultdict(int)
print(d['a'])
'''
>>> 0
'''

 

defaultdict에서는 미리 선언하지 않은 key에 해당하는 값에 값을 더해주는 것도 가능합니다.

d = defaultdict(int)
d['a'] += 5
print(d['a'])
'''
>>> 5
'''