python - 生成给定文件夹的文件名列表

Question

文件夹中保存有文档列表，我需要将文件名及其路径保存在纯文本文件中，例如

/document/file1.txt
/document/file2.txt
...

我的问题是如何迭代文件夹并提取每个文件的路径信息。谢谢。

Answer 1

你可以尝试这样的事情。

import os
output_file = open(filename, 'w')
for root, dirs, files in os.walk(".", topdown=False):
    for name in files:
        f = print(os.path.join(root, name))
        output_file.write(f)
    for name in dirs:
        print(os.path.join(root, name))
output_file.close()

您还可以使用listdir()方法

file = open(filename, 'w')
    for f in os.listdir(dirName):
        f = os.path.join(dirName, f)
        file.write(f)
 file.close()