Python:将一个目录及其子目录下的所有文件名写入一个文本文件

Question

我的问题如下。我想列出我的目录及其子目录中的所有文件名，并将该输出打印在一个 txt 文件中。现在这是我到目前为止的代码:

import os

for path, subdirs, files in os.walk('\Users\user\Desktop\Test_Py'):
   for filename in files:
     f = os.path.join(path, filename)
     a = open("output.txt", "w")
     a.write(str(f)) 

这列出了文件夹中文件的名称(有 6 个)，但每个新文件都会覆盖旧文件，因此在任何给定时间 output.txt 文件中只有一个文件名。如何更改此代码，以便它在 output.txt 文件中写入所有文件名？

Answer 1

不要在 for 循环中打开文件。在你的 for 循环之前打开它

像这样

import os

a = open("output.txt", "w")
for path, subdirs, files in os.walk(r'C:\Users\user\Desktop\Test_Py'):
   for filename in files:
     f = os.path.join(path, filename)
     a.write(str(f) + os.linesep) 

或者使用上下文管理器(更好的做法):

import os

with open("output.txt", "w") as a:
    for path, subdirs, files in os.walk(r'C:\Users\user\Desktop\Test_Py'):
       for filename in files:
         f = os.path.join(path, filename)
         a.write(str(f) + os.linesep)