将文件夹中的文件路径统计到一个txt文件的代码实现

在数据处理的时候，我们有时候需要将某个文件夹中的文件路径统计到一个txt文件中；比如，在YoloV5数据集获取中，我们需要将图像数据集中的图像路径统计到一个txt文件中，供程序根据路径进行图像查找。以下为实现代码示例，供参考：

import os
'''
file_path:需要统计的文件所在文件夹的路径
txt_path:生成的txt文件的存放路径
txt_name:生成的txt文件的命名
file_name_type:需要统计的文件夹的后缀
txt_name_type:需要统计的文件夹在txt文件中路径的后缀
'''
def filepath_generate_txt(file_path,txt_path,txt_name='name_txt.txt',file_name_type='.jpg',txt_name_type='.jpg'):
    # 1.新建一个txt格式文件夹
    txt_file=open(os.path.join(txt_path,txt_name),'w')
    # 2.获得文件下内的所有文件
    filename_list=os.listdir(file_path)
    # 3.遍历文件夹内数据并写入到txt文件中
    count=0
    for filename in filename_list:
        # 将文件夹名字和后缀分开，如'image.jpg'分为'image'和'.jpg'
        file_name=filename.split(file_name_type)[0]
        txt_write_name=os.path.join(file_path,file_name+txt_name_type)+'\n' #生成写入内容
        txt_file.write(txt_write_name) #将文件路径写入到txt文件中
        count+=1
    print('%d file turn complete'%count)
    txt_file.close()

当设定好相应的参数后，调用函数，得到效果如下所示：