(第1坑)查找文件

2020-10-15  本文已影响0人  _简姑娘_

题目:找出指定文件夹中的所有以txt结尾的文件,包括所有嵌套的子文件夹。

import os
import pprint

def search_file(dst_path):
    txt_list = []
    for root,dirs,files in os.walk(dst_path):

        for file in files:
            if file.endswith(".txt"):
                txt_list.append(file)

    return txt_list

if __name__ == "__main__":
    file_path =  "E:\\test"
    pprint.pprint(search_file(file_path))
上一篇下一篇

猜你喜欢

热点阅读