Question

我编写了一个简单的生成器函数，该函数接受一个可能包含子列表的列表，并尝试展平该列表：

所以[1，[2，3]，4，[5，[6，7]，8]]应该产生1,2,3,4,5,6,7,8

如果我只想打印出值（而不是生成器），它看起来像这样，并且可行：

#  Code A
def flatten_list_of_lists(my_list):
    for element in my_list:
        if isinstance(element, list):
            flatten_list_of_lists(element)
        else:
            print(element)

my_list = [1, [2, 3], 4, [5, [6, 7], 8]]
flatten_list_of_lists(my_list)

这将按预期打印出1,2,3,4,5,6,7,8

但是，当我将代码更改为此：

#  Code B
def flatten_list_of_lists(my_list):
    for element in my_list:
        if isinstance(element, list):
            flatten_list_of_lists(element)
        else:
            yield element

for i in flatten_list_of_lists(my_list):
    print(i)

这只是将打印转换为成品，程序仅打印出1,4。

我将在下面粘贴有效的代码。但是我想知道为什么以前的代码不起作用？如果代码A正确地“打印”出数字，为什么代码B不能正确地“屈服”数字？

好像我对生成器如何使用递归有基本的误解。

此代码实际上有效：

#  Code C
def flatten_list_of_lists_v2(my_list):
    for element in my_list:
        if isinstance(element, list):
            for sub_element in flatten_list_of_lists_v2(element):
                yield sub_element
        else:
            yield element

l = []
for element in flatten_list_of_lists_v2(my_list):
    print(element)

然后打印出1,2,3,4,5,6,7,8

仅一点点背景，我就看完了这段视频： https://www.youtube.com/watch?v=LelQTPiH3f4

在其中，他解释了在设计生成器时，只需将要打印的打印件放到要打印的位置，看看是否获得正确的结果，然后将打印件切换为打印即可。所以我想他的建议在所有情况下都行不通，我只想了解原因。

Answer 1

一个简单的错误-

def flatten_list_of_lists(my_list):
    for element in my_list:
        if isinstance(element, list):
            # add yield from
            yield from flatten_list_of_lists(element)
        else:
            # yield, not print
            yield element

my_list =  [1, [2, 3], 4, [5, [6, 7], 8]]

for e in flatten_list_of_lists(my_list):
  print(e)

输出

为什么此生成器（使用递归）不能产生直观的结果

1 个答案: