python遍历文件夹下所有的文件并返回

遍历文件夹下所有的文件，拼接成绝对路径，组成list返回

import os


def get_all_file_in_dir(dir_path):
    '''
    获取目录下的所有文件
    :return:
    '''
    file_name_list = []
    for root, dirs, files in os.walk(dir_path):
        if files:
            for name in files:
            
				# 此处可以增加文件名称过滤条件， 比如 -----
				# 跳过所有 ~$ 开头的文件
                # if name.startswith(('~$',)):
                #     continue
                # ---------------------------------------

                file_name = '{0}/{1}'.format(root, name).replace('\\', '/')
                file_name_list.append(file_name)
    return file_name_list


if __name__ == '__main__':
    img_list = get_all_file_in_dir(r'D:\机器学习数据集\celebA')
    print(img_list)