[python] defaultdict

📗 defaultdict

defaultdict 는 dict 의 서브 클래스이다. 

이름에서 유추할 수 있듯이 기본값(default)를 설정할 수 있는 좋은 친구이다.

 

🔵 defaultdict

이러한 번거로움을 없애주는게 바로 defaultdict이다.

defaultdict는 처음 선언할 때 다양한 기본값을 지정할 수 있다.

from collections import defaultdict

my_defaultdict = defaultdict(int)
print(my_defaultdict)  # defaultdict(<class 'int'>, {})
my_defaultdict = defaultdict(dict)
print(my_defaultdict)  # defaultdict(<class 'dict'>, {})
my_defaultdict = defaultdict(list)
print(my_defaultdict)  # defaultdict(<class 'list'>, {})

대표적으로 사용되는 기본값들이다. 각각 defaultdict의 클래스 타입이 지정되는 것을 볼 수 있다.

int로 설정하면 기본값이 0, dict나 list는 비어있는 dict {}나 list []가 된다.

 

🔹존재하지 않은 값을 호출 할때

my_defaultdict = defaultdict(int)
print(my_defaultdict['a'])  # 0
print(my_defaultdict)  # defaultdict(<class 'int'>, {'a': 0})

KeyError가 발생하지 않고 디폴트값이 0으로 들어간 것을 볼 수 있다.

 

🔹값을 더해줄 때

my_defaultdict = defaultdict(int)
my_defaultdict['a'] += 99
print(my_defaultdict)  # defaultdict(<class 'int'>, {'a': 99})

키값이 존재하지 않아도 값을 더해줄 수 있다.

 

🔹 리스트에서 활용

my_defaultdict = defaultdict(list)
my_defaultdict['a'].append('리스트')
print(my_defaultdict)  # defaultdict(<class 'list'>, {'a': ['리스트']})

---

🔵 dict

우리가 흔히 쓰는 딕셔너리의 경우 없는 키를 찾는다면 KeyError가 나타난다.

my_dict = {
    'a':'b'
}
print(my_dict['c'])

# KeyError: 'c'

 

하지만 이런 경우는 get()을 사용한다면 해결 할 수 있다는 것을 알 것이다.

my_dict = {
    'a':'b'
}
print(my_dict.get('c'))

# None

이렇게 키가 없다면 에러가 발생하지 않고 None이 반환된다. 

 

그렇지만 여기서 또 문제가 있다.

만약 딕셔너리에 존재하는 키값에 값을 더하거나 없다면 새로 만들고 싶을 때는 if문으로 처리를 해야 한다.

my_dict = {
    'a':'b'
}
if my_dict.get('c'):
    my_dict['c'] += 1
else:
    my_dict['c'] = 1

print(my_dict)

# {'a': 'b', 'c': 1}