今天是发电机。我看到一个question今天想找到一种方法来递归地展平列表而不使用循环和导入。 tobias_k用以下代码回答:
def flatten(test_list):
if isinstance(test_list, list):
if len(test_list) == 0:
return []
first, rest = test_list[0], test_list[1:]
return flatten(first) + flatten(rest)
else:
return [test_list]
有没有一种创建生成器的方法(遵守规则:没有导入，循环)？
注意:它纯粹是教育性的。我知道这不是最好的主意，但不知道该怎么做。
请您参考如下方法:
A generator function是一个包含至少一个 yield statement 的函数并且没有采用表达式的 return 语句。当一个生成器函数被调用时，它返回一个生成器迭代器，当它迭代时(例如通过 for 循环，或显式地使用 next)遍历函数体，卡住它的状态并在每个 yield 语句(以及在 Python 3.3 中，yield from 语句)上将控制权返回给调用者。
Python 函数内部的流控制总是向前的；如果没有像设置当前帧 f_lineno 这样的 hack(如愚人节 goto statement 所做的那样)，控制到达较早点的唯一方法是使用循环(for 或 while)。因此，在没有循环或 yield from 的情况下，生成器迭代器可以被调用的最大次数受限于生成器函数中的 yield 语句的数量。
请注意，很容易编写返回迭代器的flatten；采用原始解决方案并编写 return iter(flatten(first) + flatten(rest)) 就可以了。但那不会是生成器迭代器，函数也不会是生成器函数。
这是一个滥用 f_lineno 来提供无循环迭代的实现。不幸的是，它必须使用 import sys:
def current_frame():
i = None
def gen():
yield i.gi_frame.f_back
i = gen()
return next(i).f_back
class Loop(object):
jump = False
def __call__(self, frame, event, arg):
if self.jump:
frame.f_lineno = self.lineno
self.jump = False
return None if event == 'call' else self
def __enter__(self):
import sys
sys.settrace(self)
current_frame().f_back.f_trace = self
self.lineno = current_frame().f_back.f_lineno
return self
def __exit__(self, exc_type, exc_value, traceback):
if exc_type is None:
self.jump = True
else:
import sys
sys.settrace(None)
current_frame().f_back.f_trace = None
return exc_type is StopIteration
def flatten(x):
if isinstance(x, list):
if x:
first, rest = flatten(x[0]), flatten(x[1:])
with Loop():
yield next(first)
with Loop():
yield next(rest)
pass
else:
yield x
