Python 3.6 : Looping through files in os. listdir() 并将其中一些写入文本文档

Question

我正在尝试循环访问一些文件并将 .txt 文件的文件名写入另一个 .txt

这段代码在找到并写入一个文件的名称后停止。

我怎样才能让它写出其余人的名字？

import os

os.chdir('/users/user/desktop/directory/sub_directory')

for f in os.listdir():
    file_name, file_ext = os.path.splitext(f)
    if file_ext == '.txt':
        with open('file_test.txt', 'r+') as ft:
            ft.write(file_name)

Answer 1

您需要以“追加”模式打开目标文件

import os

os.chdir('/users/user/desktop/directory/sub_directory')

for f in os.listdir():
    file_name, file_ext = os.path.splitext(f)
    if file_ext == '.txt':
        with open('file_test.txt', 'a+') as ft:
            ft.write(file_name)

只需将“a+”作为打开函数的第二个参数(其中“a”代表“追加”，“+”代表“如果不存在则创建”)。我建议您在写入函数中添加分隔符(如“\n”)以获得更具可读性的结果