如果然后块内收益率的奇怪行为

Question

我有一个函数，它根据标志返回生成器或列表。

然而，即使我将标志设置为list该函数仍然返回一个生成器;也不会打印标志。

我希望首先评估 yield 命令之前的打印语句。 同样，如果标志设置为 list我不希望generator块评估。

import os
def get_iterator_all_files_name(dir_path, flag):
    if flag == 'generator':
        print(flag)
        for (dirpath, dirnames, filenames) in os.walk(dir_path):
            for f in filenames:
                yield os.path.join(dirpath, f)
    elif flag == 'list':
        print(flag)
        paths = list()
        for (dirpath, dirnames, filenames) in os.walk(dir_path):
            for f in filenames:
                paths.append(os.path.join(dirpath, f))
        return paths

使用函数...

file_path = 'path/to/files'
flag = 'list'
foo = get_iterator_all_files_name(file_path, flag)
type(foo)

这会产生结果...

generator

这不是我所期望的;我期待清单。

Answer 1

如果一个函数中有单词 yield，它就是一个生成器。没有例外。在尝试迭代之前，不会评估代码。

只需在结果上调用list：

def get_iterator_all_files_name(dir_path):
    for (dirpath, dirnames, filenames) in os.walk(dir_path):
        for f in filenames:
            yield os.path.join(dirpath, f)
file_path = 'path/to/files'
foo = list(get_iterator_all_files_name(file_path))

如果您确实想保留flag功能，则可以修改函数以返回生成器。您还可以将paths设置为列表推导，这可以简化功能：

def get_iterator_all_files_name(dir_path, flag):
    if flag == 'generator':
        return (os.path.join(dirpath, f) for (dirpath, dirnames, filenames) in os.walk(dir_path) for f in filenames)
    elif flag == 'list':
        return [os.path.join(dirpath, f) for (dirpath, dirnames, filenames) in os.walk(dir_path) for f in filenames]

Answer 2

任何具有yield语句的函数都将成为生成器函数，因此，其return语句具有不同的含义(与普通函数的return语句相比(。如果你真的需要 get_iterator_all_files_name 函数有时返回生成器，有时返回列表，一种方法是这样的：

1. 再定义一个函数(比如my_gen_func(来执行您现在在if子句中正在做的事情。因此，这个新功能将是一个生成器函数(因为它将有一个yield语句(。
2. 在 get_iterator_all_files_name 函数中，修改 if 子句，只调用 my_gen_func ，并返回其返回值。(现在，你的get_iterator_all_files_name函数不再是生成器函数，而只是一个普通函数，因为它没有yield语句(
3. 您的elif子句可以保持不变。