Module itertools

Mô-đun itertools trong Python là một bộ công cụ mạnh mẽ, nhanh và tiết kiệm bộ nhớ để làm việc với các iterator. Mô-đun này cung cấp một tập hợp các hàm tạo iterator để lặp và xử lý dữ liệu hiệu quả. Hãy cùng tìm hiểu một số hàm hữu ích nhất trong mô-đun này.

Nhập itertools

import itertools

Iterator vô hạn

Các iterator này sẽ tiếp tục vô hạn trừ khi bạn dừng chúng:

• count(start, step): Đếm tăng vô hạn từ một giá trị bắt đầu
• cycle(iterable): Lặp qua một đối tượng iterable vô hạn
• repeat(elem, [n]): Lặp lại một phần tử n lần (vô hạn nếu không chỉ định n)

Iterator kết thúc khi chuỗi đầu vào ngắn nhất kết thúc

• chain(*iterables): Nối nhiều đối tượng iterable với nhau
• islice(iterable, start, stop[, step]): Cắt một iterator
• zip_longest(*iterables, fillvalue=None): Tương tự zip(), nhưng tiếp tục cho đến khi đối tượng iterable dài nhất được duyệt hết

Iterator tổ hợp

• product(*iterables, repeat=1): Tích Descartes của các đối tượng iterable đầu vào
• permutations(iterable, r=None): Tất cả các hoán vị có thể có độ dài r
• combinations(iterable, r): Các tổ hợp độ dài r theo thứ tự đã sắp xếp

Ví dụ sử dụng

import itertools

# Sử dụng count
for i in itertools.count(10, 2):
    if i > 20:
        break
    print(i)  # In ra 10, 12, 14, 16, 18, 20

# Sử dụng cycle
colors = itertools.cycle(['red', 'green', 'blue'])
for _ in range(6):
    print(next(colors))  # In ra red, green, blue, red, green, blue

# Sử dụng chain
numbers = itertools.chain([1, 2, 3], [4, 5, 6])
print(list(numbers))  # In ra [1, 2, 3, 4, 5, 6]

# Sử dụng combinations
letters = 'ABCD'
print(list(itertools.combinations(letters, 2)))
# In ra [('A', 'B'), ('A', 'C'), ('A', 'D'), ('B', 'C'), ('B', 'D'), ('C', 'D')]

Mô-đun itertools còn cung cấp nhiều hàm khác, mỗi hàm được thiết kế để giải quyết hiệu quả các vấn đề cụ thể về phép lặp. Khi sử dụng những công cụ này, bạn có thể viết code ngắn gọn và hiệu quả hơn khi làm việc với iterator và các đối tượng iterable trong Python.