【每日一坑 4】 查找文件

HAO

1. # -*- coding: utf-8 -*-
   
2. 
   
3. import os
   
4. from sys import argv
   
5. 
   
6. filename, path = argv
   
7. 
   
8. txt_files = []
   
9. 
   
10. if os.path.isdir(path):
    
11.     for dir_path, dir_names, file_names in os.walk(path):
    
12.         for file_name in file_names:
    
13.             if file_name.endswith('.txt'):
    
14.                 txt_files.append(file_name)
    
15. 
    
16. 
    
17. print txt_files

复制代码

HAO

fl0w 发表于 2013-12-19 21:39
这样不能找出嵌套的子文件夹中的以.txt结尾的文件

哈哈。谢谢提醒。