python獲取指定目錄下所有文件名列表的方法

本文實例講述了python獲取指定目錄下所有文件名列表的方法。分享給大家供大家參考。具體實現(xiàn)方法如下：
這里python代碼實現(xiàn)獲取文件名列表的功能，可以指定文件中包含的字符，方便提取特定類型的文件名列表：
    
# -*- coding: utf-8 -*-
#~ #------------------------------------------------------------------
#~ module:wlab
#~ Filename:wgetfilelist.py
#~ Function :
#~ def IsSubString(SubStrList,Str)
#~ def GetFileList(FindPath,FlagStr=[]):
#~ 功能:讀取指定目錄下特定類型的文件名列表
#~ Data: 2013-08-08,星期四
#~ Author:吳徐平
#~ Email:wxp07@qq.com
#~ #------------------------------------------------------------------
#~ #------------------------------------------------------------------
def IsSubString(SubStrList,Str):
 '''''
 #判斷字符串Str是否包含序列SubStrList中的每一個子字符串
 #>>>SubStrList=['F','EMS','txt']
 #>>>Str='F06925EMS91.txt'
 #>>>IsSubString(SubStrList,Str)#return True (or False)
 '''
 flag=True
 for substr in SubStrList:
  if not(substr in Str):
   flag=False
 return flag
#~ #----------------------------------------------------------------------
def GetFileList(FindPath,FlagStr=[]):
 '''''
 #獲取目錄中指定的文件名
 #>>>FlagStr=['F','EMS','txt'] #要求文件名稱中包含這些字符
 #>>>FileList=GetFileList(FindPath,FlagStr) #
 '''
 import os
 FileList=[]
 FileNames=os.listdir(FindPath)
 if (len(FileNames)>0):
  for fn in FileNames:
   if (len(FlagStr)>0):
    #返回指定類型的文件名
    if (IsSubString(FlagStr,fn)):
     fullfilename=os.path.join(FindPath,fn)
     FileList.append(fullfilename)
   else:
    #默認直接返回所有文件名
    fullfilename=os.path.join(FindPath,fn)
    FileList.append(fullfilename)
 #對文件名排序
 if (len(FileList)>0):
  FileList.sort()
 return FileList

可以使用pip在線安裝wlab
?
1
    
pip install wlab

還是給個圖吧：

希望本文所述對大家的Python程序設計有所幫助。