Python: collections 활용하기

namedtuple

from collections import namedtuple


person_info = namedtuple("info", "name age address")
info = person_info("kim", 12, "111-11")

• info.name, info.age, info.address와 같이 접근 할 수 있다.
• name, age, address를 인덱스 0 ~ 2(ex: info[0])로도 접근할 수 있다.

defaultdict

딕셔너리를 사용하다 보면, 어떤 키에 대한 값이 없는 경우에 대한 처리를 해야하는 경우가 있다.

딕셔너리에 대한 초기값을 지정하고자 할 때 defualtdict을 사용할 수 있다.

def cnt_word(words):
    cnt = {}
    for word in words:
        if word not in cnt:
            cnt[word] = 0
        cnt[word] += 1
    return cnt

• 위의 코드는 딕셔너리에 해당 word가 없으면 0으로 초기화 해주고, 아닐 경우 +1을 하여 카운트 한다.

from collections import defaultdict


def cnt_word(words):
    cnt = defaultdict(int)
    for word in words:
        cnt[word] += 1
    return cnt

• int()의 반환 값은 0 이므로, 딕셔너리의 초기 값은 0으로 설정된다.
• 따라서 위와 같이 if 문을 통해 초기화 과정을 진행하지 않아도 된다.

deque

스택과 큐의 기능을 모두 가진 객체로 사용하는 메서드에 따라 기능을 달리 사용할 수 있다.

from collections import deque

# stack로 사용
# 입/출력 우
# FIFO
dq = deque('1234')
dq.append('5')
dq.pop()

# queue로 사용
# 입력 우, 출력 좌
# LIFO
dq.append('5')
dq.popleft()

• 값을 추가하거나 삭제할 때, append, pop을 사용하면 스택 처럼 사용할 수 있다.
• 값을 추가하거나 삭제할 때, append, leftpop을 사용하면 큐 처럼 사용할 수 있다.

Counter

리스트나 스트링에서의 각 원소들을 카운트할 때 유용하게 사용할 수 있다.

from collections import Counter


nums = [1, 3, 5, 9, 1, 2, 3, 5, 12, 15, 19, 5]
# count가 높은 수 부터 딕셔너리로 반환됨
print(Counter(nums).most_common())
# count가 가장 높은 수만 출력
print(Counter(nums).most_common(1))

• Counter의 경우 딕셔너리의 확장이기에, 딕셔너리에 사용하는 메소드를 사용할 수 있다.
• 만약 Counter에서 최대 값의 수를 반환하고자 한다면 Counter(nums).most_common()[0][1]로 사용한다.

---