如何在 Python 中有效地對迭代器進行分塊？

以區塊的形式迭代迭代器

考慮一個場景，您擁有一個迭代器l = [1, 2, 3, 4 , 5, 6 , 7] 並希望將其劃分為大小為3 的區塊，從而產生一個包含[[1, 2, 3], [4, 5, 6], [7]].

itertools 解決方案

Python 標準庫在itertools 模組中提供了一個方便的解決方案：

import itertools

def grouper(iterable, n, *, incomplete='fill', fillvalue=None):
    """Collect data into non-overlapping fixed-length chunks or blocks."""

    args = [iter(iterable)] * n
    if incomplete == 'fill':
        return zip_longest(*args, fillvalue=fillvalue)
    elif incomplete == 'strict':
        return zip(*args, strict=True)
    elif incomplete == 'ignore':
        return zip(*args)
    else:
        raise ValueError('Expected fill, strict, or ignore')

chunks = list(grouper(l, 3))  # Output: [[1, 2, 3], [4, 5, 6], [7]]

標準庫增強版本

在itertools 配方的最新版本中引入的批次專用函數，精確地滿足了這種分塊要求：

from itertools import batched

chunks = list(batched(l, 3))  # Output: [[1, 2, 3], [4, 5, 6], [7]]

非通用解決方案

特別是對於序列，一種更簡單的方法可以有效地處理最後的區塊是：

chunks = [my_list[i:i + chunk_size] for i in range(0, len(my_list), chunk_size)]

以上是如何在 Python 中有效地對迭代器進行分塊？的詳細內容。更多資訊請關注PHP中文網其他相關文章！