我有一个生成器函数,我想从另一个函数调用它并返回获得的生成器。我可以在这里看到两种方法 -
请注意,以下函数是用于说明目的的简单虚拟函数。请不要想出更好的方法来实现这些功能本身。
方法一def fun_a(n):
for i in range(n):
yield i+10
def fun_b(n):
if n < 0:
yield None
return
yield fun_a(n)
并用它list(list(fun_b(10))[0])
来获得[10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
def fun_b(n):
if n < 0:
yield None
return
for i in fun_a(n):
yield i
然后list(fun_b(10))
可以给我[10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
。
虽然方法 1 看起来不错,但我不想返回列表列表,因为在其他情况下我会返回列表并且我不想弄乱我的代码。方法2只是效率低下。
处理这种情况的真正好方法是什么?