神奇的扁平化方法是如何工作的？

Question

我正在读“开始python，从新手到专业人士”，其中有一个神奇的扁平化函数，这让我很困惑

def flatten(nested):
    try:
        for sublist in nested:
            for element in flatten(sublist):
                yield element
    except TypeError:
        yield nested

我知道收益率返回一个元素。假设我有一个列表L = [ 1, [[2]] ]。 我用这个flatte（）来称呼它，比如：

L = [ 1, [[2]] ]
for i in flatten(L):
    print i

1
2

我真的很困惑，当我们调用for循环时，我们是否会触发flatte（），我们看到第一个元素1，它无疑会导致try块中的TypeError，except块中返回的nested变量不应该是整个列表[1, [[2]] ]？为什么它会返回1

Answer 1

我想我现在明白这里发生了什么

def flatten(nested):
    try:
        for sublist in nested:
            for element in flatten(sublist):
                yield element
    except TypeError:
        yield nested

我们仍然有L = [1, [[2]]]，我们称之为flatten(L)

现在第一个元素是1，现在我们移动到for element in flatten(1)

当我们调用flatte（）并遇到TypeError时，我们将得到nested变量，在这里，它是1

所以我们得到了1

原来的L现在只剩下[[2]]

正如我们所调用的for循环：for i in flatten(L)，事情将继续进行

现在我们将转到for element in flatten([[2]])，是的，我们在它下面有一个yield element，但是我们必须先执行for element in flatten([[2]])，然后看看发生了什么

现在我们将转到for element in flatten([2])，直接转到它下面的yield element是一种诱惑。但是，等一下，我们下次再去找平

现在我们移动到flatten(2)，它将像flatte（1）一样引发TypeError

所以这次我们得到了2

事情解释清楚了

Answer 2

输入一个打印语句来查看发生了什么是很有帮助的。例如

#! /usr/bin/env python

def flatten(nested, depth=0):
    print "%d: %r" % (depth, nested)
    try:
        for sublist in nested:
            for element in flatten(sublist, depth+1):
                yield element
    except TypeError:
        yield nested


def main():
    L = [ 1, [[2]] ]
    for i in flatten(L):
        print "output %s\n" % i


if __name__ == '__main__':
    main()

输出

0: [1, [[2]]]
1: 1
output 1

1: [[2]]
2: [2]
3: 2
output 2

Answer 3

也许不使用yield重写函数可以让您更容易理解：

def flatten(nested):
    res = []
    try:
        for sublist in nested:
            for element in flatten(sublist):
                res.append(element)
    except TypeError:
        res.append(nested)
    return res

用最基本的术语来说，这就是yield所做的。它从函数一次“发送”一个值。有关更完整的解释，请参见What does the yield keyword do in Python?的答案