[Python] collections 모듈

💡collections 모듈💡

collections 모듈은 파이썬 범용 내장 컨테이너 dict, list, set 및 tuple에 대한 대안을 제공하는 특수 컨테이너 데이터형을 제공하는 모듈이다. 대표적으로 Counter, defaultdict, deque 등이 있다.

💡Counter 클래스💡

Counter 클래스 특징

(1) value값에 0 또는 음수 값도 허용
(2) 생성 시 O(n)의 시간 복잡도를 가짐
(3) 누락된 항목에 대해 KeyError가 발생하지 않고, 대신 0을 반환

(1) Counter(n) 함수

n(빈 객체, iterable한 객체, mapping된 객체, keyword arguemnt)을 dictionary화하여 반환한다.

from collections import Counter

c1 = Counter()

>>> print(c1)
Counter()

c2 = Counter('Hello')

>>> print(c2)
Counter({'l': 2, 'H': 1, 'e': 1, 'o': 1})

c3 = Counter({'a' : 4, 'b' : 3})

>>> print(c3)
Counter({'a': 4, 'b': 3})

c4 = Counter(a = 1, b = 3)

>>> print(c4)
Counter({'b': 3, 'a': 1})

(2) elements() 함수

value의 값이 1이상인 key값을 반환한다.

from collections import Counter

c = Counter(a = 3, b = 2, c = 0)

>>> print(list(c.elements()))
['a', 'a', 'a', 'b', 'b']

(3) most_common([n])

value가 큰 key부터 내림차순으로 n개만큼 반환한다. n이 생략되거나 None이라면, 모든 key를 내림차순으로 반환한다. value가 같은 key는 처음 순서를 유지한다.

from collections import Counter

c = Counter(a = 3, b = 2, c = 0)

>>> print(c.most_common())
[('a', 3), ('b', 2), ('c', 0)]

(4) total()

전체 value의 개수를 계산하여 반환한다.

from collections import Counter

c = Counter(a = 3, b = 2, c = 0)

>>> print(c.total())
5

💡defaultdict 클래스💡

defaultdict 클래스 특징

key를 지정할 때 값을 주지 않을 경우, default값을 지정할 수 있음

(1) defaultdict(n) 함수

n(자료형, 자료구조)을 기본인자로 하는 dictionary생성

from collections import defaultdict

dic1 = defaultdict(int)

>>> print(dic1[0])
0

dic2 = defaultdict(float)

>>> print(dic2[0])
0.0

dic3 = defaultdict(list)

>>> print(dic3[0])
[]

💡deque 클래스💡

deque 클래스 특징

(1) 양 끝 element에 대해 append, pop 가능
(2) stack과 queue를 동시에 사용 가능
(3) list에 비해 속도가 빠름

(1) append(item)

queue의 맨 뒤에 item을 추가한다.

from collections import deque

queue = deque()
queue.append(3)
queue.append(4)

>>> print(queue)
deque([3, 4])

(2) appendleft(item)

queue의 맨 앞에 item을 추가한다.

from collections import deque

queue = deque()
queue.append(3)
queue.appendleft(4)

>>> print(queue)
deque([4, 3])

(3) pop()

queue의 맨 뒤의 원소를 꺼낸다.

from collections import deque

queue = deque([1,2,3])
queue.pop()

>>> print(queue)
deque([1, 2])

(4) popleft()

queue의 맨 앞의 원소를 꺼낸다.

from collections import deque

queue = deque([1,2,3])
queue.popleft()

>>> print(queue)
deque([2, 3])

(5) rotate(n)

n만큼 원소를 회전시킨다. 시계방향 회전은 양수, 반시계방향은 음수를 n에 대입한다.

from collections import deque

queue = deque([1, 2, 3, 4, 5])
queue.rotate(2)

>>> print(queue)
deque([4, 5, 1, 2, 3])