遍历目录中的txt文件，保存文件名

import os

def traverse_directory(directory):
    txt_files = []  # 保存txt文件名的列表

    # 遍历目录中的所有文件和子目录
    for root, dirs, files in os.walk(directory):
        for file in files:
            # 判断文件是否为txt文件
            if file.endswith(".txt"):
                txt_files.append(file)  # 将文件名添加到列表中

    return txt_files

# 调用函数遍历目录并保存txt文件名
directory = "/path/to/directory"  # 替换为实际的目录路径
txt_files = traverse_directory(directory)

# 打印保存的txt文件名
for file in txt_files:
    print(file)