python - 拼图 : Recursive generator without loops/imports

Question

今天是发电机。我看到一个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]

有没有一种创建生成器的方法(遵守规则:没有导入，循环)？

注意:它纯粹是教育性的。我知道这不是最好的主意，但不知道该怎么做。

Answer 1

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