下面是一个使用Python语言遍历目录中的txt文件并保存文件名的示例代码:
import os
def save_txt_filenames(directory):
filenames = []
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith(".txt"):
filenames.append(file)
with open("txt_filenames.txt", "w") as file:
for filename in filenames:
file.write(filename + "\n")
# 调用函数,传入要遍历的目录路径
save_txt_filenames("/path/to/directory")
这个示例中,我们使用os.walk()
函数遍历指定目录下的所有文件和子文件夹。对于每个文件,我们检查文件名是否以".txt"结尾,如果是,则将文件名添加到filenames
列表中。最后,我们将filenames
列表中的文件名逐行写入到一个名为"txt_filenames.txt"的文本文件中。
你可以根据自己的需要修改代码中的目录路径和保存文件名的文件名。