参考博客链接:
python 获取当前文件夹下所有文件名 - StrongYaYa - 博客园
https://www.cnblogs.com/strongYaYa/p/7200357.html
import os
def load_files(file_dir):
"""
加载文件夹路径下的所有的文件
"""
for dirs in os.walk(file_dir):
fun_dirs = dirs[2] # 当前路径下所有子目录
for i in range(len(fun_dirs)):
# 筛选文件名的最后一个是y结尾的文件
if fun_dirs[i][-1]=='y':
file = open('E:\\python\\function"'\\'+fun_dirs[i],'r', encoding='UTF-8')
file_comm = []
# 循环读取每个文件的2-10行的信息并输出
for line in file.readlines()[1:9]:
file_comm.append(line)
print(file_comm)
# 调用函数,并传入路径参数
info=load_files("E:\\python\\function")