在Python中用块（n）迭代迭代器？

Question

在Python中用块（n）迭代迭代器？你能想出一个很好的方法（可能用itertools）将迭代器拆分成给定大小的块吗？因此l=[1,2,3,4,5,6,7]与chunks(l,3)变成一个迭代[1,2,3], [4,5,6], [7]我可以想到一个小程序来做这个，但不是一个很好的方式可能是itertools。

Answer 1

在grouper() 从配方itertools文件的食谱来靠近你想要什么：

def grouper(n, iterable, fillvalue=None):
    "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
    args = [iter(iterable)] * n    return izip_longest(fillvalue=fillvalue, *args)

不过，它会用填充值填充最后一个块。

一种不太通用的解决方案，仅适用于序列，但可以根据需要处理最后一个块

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

最后，对一般迭代器起作用的解决方案的行为符合要求

def grouper(n, iterable):
    it = iter(iterable)
    while True:
       chunk = tuple(itertools.islice(it, n))
       if not chunk:
           return
       yield chunk

Answer 2

这适用于任何可迭代的。它返回发电机的发电机（充分灵活）。我现在意识到它与@reclosedevs解决方案基本相同，但没有绒毛。无需try...except为StopIteration传播了起来，这是我们想要的。

在next(iterable)需要调用，以提高StopIteration当迭代是空的，因为islice将永远继续下去产卵空的发电机，如果你让它。

它更好，因为它只有两行，但很容易理解。

def grouper(iterable, n):
    while True:
        yield itertools.chain((next(iterable),), itertools.islice(iterable, n-1))

请注意，它next(iterable)被放入元组中。否则，如果next(iterable)它本身是可迭代的，那么itertools.chain就会把它弄平。感谢Jeremy Brown指出这个问题。