【每日一坑 4】 查找文件

scz_000

1. import os,fnmatch
   
2. def find_txt(root,patterns='*.txt'):#定义主函数，接受目录与所要匹配的拓展名为参数
   
3.     list_of_txt=[]#列表储存txt文件名
   
4.     for dirname,subdir,filenames in os.walk(root):#使用os。walk（遍历包括子目录下所有文件），生成三组列表，第一组为当前目录地址，第二组为当下前目录下的文件夹名（没有则返回一个空列表），第三组为当前目录的所有文件
   
5.         for name in filenames:#遍历每组文件名
   
6.             if fnmatch.fnmatch(name,patterns):#用fnmatch。fnmatch寻找匹配的文件
   
7.                 list_of_txt.append(name)#加入列表
   
8.     return list_of_txt
   
9. if __name__=='__main__':
   
10.     dir=raw_input('directory:')
    
11.     txtfiles=find_txt(dir)
    
12.     print txtfiles

复制代码