Toccata in Nowhere.

Python函数工具:获取文件列表,可选前缀后缀约束

2020.02.24

通过给定的目录获取文件夹的目录,返回目录下所有文件名列表。

库依赖

import os

参数

参数 说明 默认 类型
basis_dir 工作目录 默认当前工作目录./ 可选参数
begin 筛选文件前缀 默认不筛选 可选参数
end 筛选文件后缀 默认不筛选 可选参数

代码

def get_file_list(basis_dir="./", begin="", end=""):
	path_list = os.listdir(basis_dir)
	list_final = []
	for partial in path_list:
		if begin and end:
			if partial[:len(begin)] == begin and partial[-len(end):] == end:
				list_final.append(partial)
				
		elif end:
			if partial[-len(end):] == end:
				list_final.append(partial)
		
		elif begin:
			if partial[:len(begin)] == begin:
				list_final.append(partial)
				
		else:
			list_final.append(partial)
			
	return list_final

使用例

获取所有.pdf文件,循环遍历文件名

file_list = get_file_list("./", end=".pdf")

for file_name in file_list:
	## operation here