获取文件夹下的文件列表，并导出为txt文件

#! /usr/bin/env python

# -*- coding: utf-8 -*-

# __author__ = "wxf"

# Email: [email protected]

# Date: 2018/8/20

import os

def get_file_path(file_dir):

    file_path_list = []

    for root, dirs, files in os.walk(file_dir):

        # print(u'当前目录路径', root)  # 当前目录路径

        # print(u'当前路径下所有子目录', dirs)  # 当前路径下所有子目录

        # print(u'当前路径下所有非目录子文件', files)  # 当前路径下所有非目录子文件

        for file_name in files:

            file_path_list.append(os.path.join(root, file_name))

            # print(os.path.join(root, file_name))

        pass

    pass

    return file_path_list

pass

if __name__ == '__main__':

    file_dir_base = 'G:/wxf/deep_Learn/sphereface/train/data/'

    data_set_name = 'lfw'

    data_set_dir = file_dir_base + data_set_name

    f1 = open(os.path.join(file_dir_base, data_set_name + '.txt'), 'w')

    data_set_path_list = get_file_path(data_set_dir)

    for t in data_set_path_list:

        f1.write(t + '\n')

    pass

    f1.close()

pass