1,循环文件内的目录
def file_load(dir, Filelist):
# 读取数据文件路径名字 # new_dir = dir
if os.path.isfile(dir):
Filelist.append(dir)
elif os.path.isdir(dir):
for s in os.listdir(dir):
l = []
l.append(dir)
l.append(s)
new_dir = '/'.join(l)
# new_dir = os.path.join(dir, s)
file_load(new_dir, Filelist)
return Filelist
2,打开文件
因为我只需要打开文件直接读取数据,使用open比较快,如有其它需求使用Python封装函数read()
def read_files(file):
# 读取数据文件内容
gpul_name = file.split('/')[-1].split('-')[0]
demand_data = {}
demand_data1 = {}
f = open(file)
content = f.readlines()
for line in content:
line = line.replace('\n', '').split(",")
if line[0] != "Time":
demand_data[line[0]] = (line[3], line[4], line[5])
demand_data1[gpul_name] = demand_data
return demand_data1
3,入口
if __name__ == '__main__':
road = file_load('E:/yuce/new/zzz/sj', [])