Python:遍历当前路径下所有目录，并以列表形式返回指定格式文件

我们是轮子的搬运工

import os
def find_map_path():
    path_list = [path for path in os.listdir(r'./') if os.path.isdir(path)]
    file_list = []
    for path in path_list:
        for root, dirpath, files in os.walk(path):
            for file in files:
                suffix = file.split('.')
                if 'txt' in suffix:                              #后缀为.txt的则进入计算
                    file_list.append(os.path.join(root, file))
    return file_list

def main():
    map_lists = find_map_path()    #获取map路径
    print(map_lists)
if __name__ == '__main__':
    main()