使用生成器函数迭代可迭代对象

时间:2019-02-19 20:55:52

标签: python iterator generator yield iterable

我正面临这个问题,我希望你们中的一些人能有所帮助:

  

编写一个接受可迭代数和正数n的函数。该函数返回一个新的迭代器,该迭代器以长度为n的元组给出原始值。如果最后一个元组需要,用'None'填充缺失值。

例如:

for x in bunch_together(range(10),3): print(x)

返回值是

(0, 1, 2) (3, 4, 5) (6, 7, 8) (9, None, None)

这是我到目前为止提出的:

def bunch_together(iterable,n):
    tup = tuple()
    for item in iterable:
        for i in range(n):
            tup += (i,)
        yield tup

但这显然不起作用,因为我根本没有考虑范围(截至目前的输出如下所示:

(0, 1, 2)
(0, 1, 2, 0, 1, 2)
(0, 1, 2, 0, 1, 2, 0, 1, 2)
...#(goes on)

我可以创建一个生成迭代器的生成器(就像在构建一个由init iter和next组成的类时一样) 谢谢您的帮助!

1 个答案:

答案 0 :(得分:0)

尝试在for循环中初始化元组

def bunch_together(iterable,n):

    for k in range(0,len(iterable),n):
        tup = tuple()
        for i in range(k,k+n):
            tup += (iterable[i] if i<len(iterable) else None,) # condition to check overflow
        yield tup

for x in bunch_together(range(10),3): 
    print(x)

输出

(0, 1, 2)
(3, 4, 5)
(6, 7, 8)
(9, None, None)