import os
import pandas as pd
import numpy as np
#传入参数:file_dir-文件所在路径,ext-目标文件后缀
def file_list(file_dir,ext='.txt'):
#将目标文件名保存为list
L=list(filter(lambda filename: os.path.splitext(filename)[1] == ext,os.listdir(file_dir)))
#初始化中间变量
result=pd.DataFrame()
tt=pd.DataFrame()
#print(L)
for i in L:
#print(i)
tdate=i[0:4] #取文件名前4位的日期值
n=i.find('ym') #找到关键字’ym‘的index
tag=i[n-1:n+1] #截取目标字符串
f=pd.read_csv(i,header=None) #读入文件
#定义拼接函数
def con(x):
return tdate+','+str(x)+','+tag
a=f[0].apply(con) #应用拼接函数到数据框的列
tt=pd.concat([result,a]) #结果整合
result=tt #更新result
tt.to_csv("rr.csv",index=False,header=False,encoding='GBK') #结果输出至csv文件
return tt #屏幕输出,可注释
file_dir="/Users/ym/Documents/test" #定义目录路径
file_list(file_dir) #调用函数