获取指定后缀的文件列表

获取指定后缀的文件列表

import os
def get_file_list(data_path,endwith):
    file_list = []
    for root, dirs, files in os.walk(data_path):
        for file in files:
            if os.path.splitext(file)[1] == '.'+endwith:
                file_list.append(os.path.join(root,file))
    return file_list
if __name__=='__main__':
    #获取当前目录下的mp3文件
    print(get_file_list('.','mp3'))