[ CLEAN CODE ] Clean Python, Dictionary 활용하기

Dictionary

       - 파이썬에서 Dictionary를 활용할 때, 어떤 Key에 대한 Value를 처리해야 하는 경우가 상당히 많이 존재합니다.

 

EX>

def counterLetters(word):
    counter = {}
    for letter in word:
        if letter not in counter:
            counter[letter] = 0
        counter[letter] += 1
     
    return counter

* 위 코드의 문제는 letter가 counter내에 존재하지 않으면, 초기 세팅을 해주는 코드입니다.
  하지만, 위 코드의 문제는 가독성이 상당히 떨어진다는 상당한 문제점이 존재합니다.

Dict.setdefault

       - Key와 Value를 인자로 받는 Dictionary의 Method입니다.

          원리는 setdefault는 Key가 존재한다면,  Key를 반환하고, 없다면, Values를 반환합니다.

 

위의 코드를 조금 수정해보자면, 

def counterletters(word):
    counter = {}
    for letter in word:
        counter.setdefault(letter, 0)
        counter[letter] += 1
    return counter

* 가독성이 향상되지만, 무조건적으로 setdefault가 호출된다는 문제점이 있습니다.

Collection.defaultdict

       - defaultdict은  dictionary의 기본값을 정의하고, 값이 없는 경우 에러를 출력하지 않고, 기본값을 출력합니다.

          setdefault보다 빠르다는 장점을 가지고 있습니다.

       

       - 기본값으로 value가 아닌, list, object등 객체를 지정하는 것도 가능합니다.

from collections import defaultdict

def counterLetters(word):
    counter = defaultdict(int)
    
    for letter in word:
        counter[letter] += 1
    
    return counter

 

참고

https://www.daleseo.com/python-collections-defaultdict/

[파이썬] 사전의 기본값 처리 (dict.setdefault / collections.defaultdict)

https://gomguard.tistory.com/126

[Python] Dictonary 관련 함수 - setdefault, defaultdict, counter