python - 以惯用的方式在 Python 中遍历 os.walk 中的各个文件

Question

我从我从 another stackoverflow question 得到的一些代码开始为目录树中的所有文件生成完整路径:

import os
def recursive_file_gen(mydir):
    for root, dirs, files in os.walk(mydir):
        for file in files:
            yield os.path.join(root, file)

我想添加内存，在我看来最简单的方法就是让 recursive_file_gen 返回一个列表:

def recursive_file_list(mydir):
    result = []
    for root, dirs, files in os.walk(mydir):
        for file in files:
            result.append(os.path.join(root, file))
    return result

这段代码真的很好。弄清楚发生了什么当然不难。但是使用连续的 append 操作来组合一个列表并不完全是 Pythonic。我的猜测是有一种使用 Python 语法的更好方法，可能是通过列表理解，学习它会很有启发。

我当然可以写:

def recursive_file_list(mydir):
    return list(recursive_file_gen(mydir))

并记住它。但我想知道是否有更清洁的东西。

Answer 1

怎么样？

result = [os.path.join(root, file) for root, dir, files in os.walk(mydir) for file in files]