最近碰到读取一个目录中所有的pickle文件的任务,顺便记录一下。
import os
import pandas as pd
from collections import defaultdict, Counter
import numpy as np
#调用与当前.py文件不在同一个目录下的.py文件的函数模块
import sys
sys.path.append(r'D:\PycharmProjects\A\B')
from C import *
import dill as pickle
for filename in os.listdir('./test'):#filename: file name 0_Rules.pickle
abs_path = os.path.abspath('./test') #文件夹的绝对路径
complete_filename = os.path.join(abs_path,filename) #将路径与文件名结合起来就是每个文件的完整路径
str_filename = filename.split('_') # 通过指定分隔符对字符串进行切片,返回分割后的字符串列表
iteration = str_filename[0]
df_final_Rules = LoadVariablestoPickleFile(complete_filename)#这个函数是自定义的,位置在D:\PycharmProjects\A\B\C中。用于导入pickle文件至内存
columns_list = df_final_Rules.columns.values.tolist()
参考
【1】https://blog.csdn.net/l641208111/article/details/101116809