我正面临这个问题,我希望你们中的一些人能有所帮助:
编写一个接受可迭代数和正数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组成的类时一样) 谢谢您的帮助!
答案 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)