【每日一坑 4】 查找文件

liucy

1. # coding:utf-8
   
2. import os
   
3. import fnmatch
   
4. 
   
5. def findfile(inputdir):
   
6.     txtlist = []
   
7.     for parent, dirnames, filename in os.walk(inputdir):
   
8.         for filenames in filename:
   
9.             txtlist.append(filenames)
   
10.         return fnmatch.filter(txtlist, '*.txt')
    
11. 
    
12. if __name__ == "__main__":
    
13.     inputdir1 = input("Please input your dir(like this: '/Users/lemon') : ")
    
14.     print findfile(inputdir1)

复制代码