在Python中按n个块迭代迭代器?

问题描述

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

解决方法

您能想到一种很好的方法(也许使用itertools)将迭代器拆分为给定大小的块吗?

因此,l=[1,2,3,4,5,6,7]withchunks(l,3)成为迭代器[1,3],[4,6],[7]

我可以想到一个小程序来做到这一点,但是使用itertools并不是一个好方法。