1 # -*- coding:utf-8 -*- 2 import os 3 import re 4 p1=r"([0-9][0-9][AB]).w{3}$" 5 pattern1=re.compile(p1) 6 def get_dir(zz): 7 listdir=[] 8 for filename in os.listdir('./%s'%zz): 9 listdir.append(filename) 10 # print(filename) 11 # print(type(filename)) 12 # print(listdir) 13 return listdir 14 15 def joint_b0(listdir,zz): 16 if not os.path.exists('./new/'): 17 os.makedirs('./new/') 18 fw=open('./new/%s.csv'%(zz),'a') 19 for i in listdir: 20 # print(i) 21 j=1000 22 matcher1=re.findall(pattern1,i) 23 fr=open('./%s/%s'%(zz,i)) 24 for line in fr.readlines(): 25 try: 26 # print(len(line)) 27 if(len(line)==1): 28 continue 29 else: 30 fw.write(matcher1[0]+'_%s'%j+','+line.strip(' ')+' ') 31 j=j+1 32 except: 33 pass 34 fr.close() 35 fw.close() 36 37 def joint_b1(listdir,zz):#单独拼接 38 if not os.path.exists('./new/'): 39 os.makedirs('./new/') 40 fw=open('./new/%s.csv'%(zz),'a') 41 for i in listdir: 42 # print(i) 43 j=1 44 matcher1=re.findall(pattern1,i) 45 fr=open('./%s/%s'%(zz,i)) 46 for line in fr.readlines(): 47 fw.write(line) 48 49 fr.close() 50 fw.close() 51 if __name__=="__main__": 52 53 # dir_list=['b0'],'b3','b4','b2_idsd','b3_idcd' 54 55 56 dir_list=['b4']#用于编号和拼接，会在new目录下生成编号号码的文件 57 for i in range(len(dir_list)): 58 listdir=get_dir(dir_list[i]) 59 # print(listdir) 60 joint_b0(listdir,dir_list[i]) 61 62 # dir_list=['stock']#用于拼接，将需要拼接的放到stock目录下，会在new目录下生成stock文件，然后根据需要修改名称 63 # for i in range(len(dir_list)): 64 # listdir=get_dir(dir_list[i]) 65 # # print(listdir) 66 # joint_b1(listdir,dir_list[i])