本文实例讲述了python获取指定目录下所有文件名列表的方法。分享给大家供大家参考。具体实现方法如下：
这里python代码实现获取文件名列表的功能，可以指定文件中包含的字符，方便提取特定类型的文件名列表：
# -*- 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
pip install wlab
还是给个图吧：
希望本文所述对大家的python程序设计有所帮助。