【每日一坑 4】 查找文件

Kris_coding

刚开始学，看前面几个的答案有点不能理解，然后搜索了 os.walk 的用法，搞了2、3个小时终于跑通了～
回帖庆祝

1. # 找出指定文件夹中的所有以txt结尾的文件，包括所有嵌套的子文件夹。
   
2. 
   
3. import os
   
4. 
   
5. 
   
6. def findfile(path,rule = ".txt"):
   
7.     txtlist = []
   
8.     for fpathe,dirs,fs in os.walk(path):        # os.walk 获取所有目录
   
9.         for f in fs:
   
10.             filename = os.path.join(fpathe,f)
    
11.             #print(filename)
    
12.             if filename.endswith(rule):         # 判断是否用 .txt 结尾
    
13.                 txtlist.append(filename)
    
14.     return txtlist
    
15. 
    
16. if __name__ == "__main__":
    
17.     b = findfile(r"/desktop")
    
18.     #print(b)
    
19.     for i in b:
    
20.         print(i)
    
21.     print('一共有个 %d 个文件'%len(b))      # len 计算 b 的长度

复制代码