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

Gerenuk:

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

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

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

斯文·马纳赫(Sven Marnach):

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

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章