一、获取指定目录下、指定后缀的文件名列表
""" 函数说明:获取指定目录下的、指定后缀的文件 例如：.xlsx、.json Parameters: path - 目录所在的路径 例如 path='D:\Python Example\Tianyancha\Data' suffix - 后缀,例如'.xlsx' Returns: input_template_All - 指定后缀的所有文件名 Author: heda3 Blog: https://blog.csdn.net/heda3 Modify: 2019-10-12 """ #参考：https://www.runoob.com/python/os-listdir.html def getFileName1(path,suffix): # 获取指定目录下的所有指定后缀的文件名 input_template_All=[] f_list = os.listdir(path)#返回文件名 for i in f_list: # os.path.splitext():分离文件名与扩展名 if os.path.splitext(i)[1] ==suffix: input_template_All.append(i) #print(i) return input_template_All
二、（升级版）获取指定目录下指定后缀的文件名列表和文件名+目录的拼接
例如：D:\Python Example\Tianyancha\Data\xx.xlsx
""" 函数说明:获取指定目录下的、指定后缀的文件名及路径+文件名的拼接 例如：.xlsx、.json Parameters: path - 目录所在的路径 例如 path='D:\Python Example\Tianyancha\Data' suffix - 后缀,例如'.xlsx' Returns: input_template_All - 指定后缀的所有文件名 xx.xlsx input_template_All_Path - 文件名和该路径的拼接 例如：D:\Python Example\Tianyancha\Data\xx.xlsx Author: heda3 Blog: https://blog.csdn.net/heda3 Modify: 2019-10-12 """ #参考：https://www.runoob.com/python/os-walk.html def getFileName2(path,suffix): input_template_All=[] input_template_All_Path=[] for root, dirs, files in os.walk(path, topdown=False): for name in files: #print(os.path.join(root, name)) print(name) if os.path.splitext(name)[1] == suffix: input_template_All.append(name) input_template_All_Path.append(os.path.join(root, name)) return input_template_All,input_template_All_Path
测试案例：
import os path='D:\Python Example\Tianyancha\Data' input_template_All1=getFileName1(path,'.xlsx') input_template_All2,input_template_All_Path2=getFileName2(path,'.xlsx')
可更该为获取当前目录：
os.path.abspath('.')#获取当前工作目录路径 介绍其它获取方法https://www.cnblogs.com/Jomini/p/8636129.html
参考文档：
【1】Python os.listdir() 方法 https://www.runoob.com/python/os-listdir.html
【2】Python os.walk() 方法 https://www.runoob.com/python/os-walk.html
【3】os.path.split(path) 把路径分割成 dirname 和 basename，返回一个元组
os.path.join(path1[, path2[, ...]]) 把目录和文件名合成一个路径
Python os.path() 模块（全） https://www.runoob.com/python/python-os-path.html
walk()方法语法格式如下：
os.walk(top[, topdown=True[, onerror=None[, followlinks=False]]])
参数
top -- 是你所要遍历的目录的地址, 返回的是一个三元组(root,dirs,files)。
root 所指的是当前正在遍历的这个文件夹的本身的地址
dirs 是一个 list ，内容是该文件夹中所有的目录的名字(不包括子目录)
files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录)
topdown --可选，为 True，则优先遍历 top 目录，否则优先遍历 top 的子目录(默认为开启)。如果 topdown 参数为 True，walk 会遍历top文件夹，与top 文件夹中每一个子目录。
onerror -- 可选，需要一个 callable 对象，当 walk 需要异常时，会调用。
followlinks -- 可选，如果为 True，则会遍历目录下的快捷方式(linux 下是软连接 symbolic link )实际所指的目录(默认关闭)，如果为 False，则优先遍历 top 的子目录。
返回值
该方法没有返回值。
