Python 扫面文件夹中的文件

 #coding=utf-8
 import os,sys
 reload(sys)
 sys.setdefaultencoding("utf-8")
 def scan_files_sub_dir(directory,prefix=None,postfix=None):
     files_list=[]
     for root, sub_dirs, files in os.walk(directory):
         for special_file in files:
             if postfix:
                 if special_file.endswith(postfix):
                     files_list.append(os.path.join(root,special_file))
             elif prefix:
                 if special_file.startswith(prefix):
                     files_list.append(os.path.join(root,special_file))
             else:
                 files_list.append(os.path.join(root,special_file))
     return files_list
 files  = scan_files_sub_dir(".",None,".txt")
 print files

 def scan_files(directory,prefix=None,postfix=None):
     files_list=[]
     for file in os.listdir(directory):
         if postfix:
                 if file.endswith(postfix):
                     files_list.append(file)
         elif prefix:
             if file.startswith(prefix):
                 files_list.append(file)
         else:
             files_list.append(file)
     return files_list
 files  = scan_files(".",None,".txt")
 print files

另一种方法:

import os
def get_file_lists(init_dir,postfix=None):
    file_lists = []
    if not os.path.exists(init_dir):
        print('%s not exist'%init_dir)
        return file_lists
    if not os.path.isdir(init_dir):
        print('%s not dir'%init_dir)
        return file_lists
    for file in os.listdir(init_dir):
        path = "%s\%s"%(init_dir,file)
        if postfix and not file.endswith(postfix) and os.path.isfile(path):
            continue
        if not os.path.isdir(path):
            file_lists.append(path)
        if os.path.isdir(path):
            file_lists.extend(get_file_lists(path,postfix))
    return file_lists

file_lists = get_file_lists(r'D:\Program Files\Java','java')

file_writer = open('list.txt','w',encoding='utf-8')
for file in file_lists:
    file_writer.write("%s\n"%file)
file_writer.close()