collections & itertools

from collections import Counter

words = ["a", "b", "a", "c", "a", "b"]
count = Counter(words)
print(count)              # Counter({'a': 3, 'b': 2, 'c': 1})
print(count["a"])         # 3
print(count.most_common(2))   # [('a', 3), ('b', 2)]  top-2

# นับตัวอักษรในข้อความก็ได้
print(Counter("banana"))  # Counter({'a': 3, 'n': 2, 'b': 1})

from collections import defaultdict

# จัดกลุ่มนักเรียนตามเกรด
students = [("Aph", "A"), ("Bee", "B"), ("Cha", "A")]
groups = defaultdict(list)        # ค่า default ของ key ใหม่ = list ว่าง
for name, grade in students:
    groups[grade].append(name)    # ไม่ต้องเช็คว่ามี key ไหม

print(dict(groups))   # {'A': ['Aph', 'Cha'], 'B': ['Bee']}

# นับด้วย defaultdict(int) ก็ได้
counts = defaultdict(int)
for ch in "banana":
    counts[ch] += 1               # key ใหม่เริ่มที่ 0 อัตโนมัติ

from collections import namedtuple

Point = namedtuple("Point", ["x", "y"])
p = Point(3, 4)
print(p.x, p.y)      # 3 4  (อ่านง่ายกว่า p[0], p[1])
print(p)             # Point(x=3, y=4)

from collections import deque

q = deque([1, 2, 3])
q.append(4)          # ต่อท้าย -> [1, 2, 3, 4]
q.appendleft(0)      # ต่อหน้า -> [0, 1, 2, 3, 4]
print(q.popleft())   # 0  (ดึงหน้า เร็ว O(1))
print(q.pop())       # 4  (ดึงท้าย)

from itertools import chain, combinations, product, count

# chain: ต่อหลาย iterable เข้าด้วยกัน
print(list(chain([1, 2], [3, 4])))        # [1, 2, 3, 4]

# combinations: เลือก r ตัวจากชุด (ไม่สนลำดับ)
print(list(combinations(["a", "b", "c"], 2)))
# [('a','b'), ('a','c'), ('b','c')]

# product: ผลคูณคาร์ทีเซียน (ทุกการจับคู่)
print(list(product([1, 2], ["x", "y"])))
# [(1,'x'), (1,'y'), (2,'x'), (2,'y')]

# count: นับไม่รู้จบ (ใช้คู่ zip/break)
for i, ch in zip(count(1), "abc"):
    print(i, ch)    # 1 a / 2 b / 3 c