1. 程式人生 > >python迭代列出某資料夾下所有檔案

python迭代列出某資料夾下所有檔案

# -*- coding: UTF-8 -*- 
# 輸出一個資料夾下所有檔案
import os
import re
def filterFileName(filename):
if re.search('.doc\w{0,1}',filename,re.I)!=None:
return filename

def getalldocfilename(path):
    docfilenames = []
    for dirpath, dirnames, filenames in os.walk(path):
        filenames = filter(filterFileName, filenames)
        filenames = map(lambda filename: os.path.join(dirpath, filename), filenames)
        docfilenames.extend(filenames)
        return docfilenames
def main():
print 'begin'
path="C:/Users/sun/Desktop"
files =getalldocfilename(path)
print files

if __name__=='__main__':
main()