通过给定的目录获取文件夹的目录,返回目录下所有文件名列表。
库依赖
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