python读取文件夹并写入读取到的文件到txt文件中

# !/usr/bin/env python
# -*- encoding: utf-8 -*-

import os


train_path = r'D:\下载\DJI_202305131522_050\train\haze'  # 第1处 需要修改为自己的路径

train_list = os.listdir(train_path)


# 输出所得到的文件名

# 打开 txt 文件，并将路径写入
with open(r"D:\文档\Temp\Tmp\trainlist.txt", 'w') as f:  # 第2处 需要修改为自己的路径
    for train_name in train_list:
        f.write(train_name + '\n')  # 第3处 需要修改为自己的路径


print('\n' + 'OK')