import os
import re
f_cg=open('dm_cg.txt','w')
f_sb=open('dm_sb.txt','w')
dir_list=[]
for file in os.listdir():
if os.path.splitext(file)[1] == '.txt':
dir_list.append(file)
dir_list.remove('dm_cg.txt')
dir_list.remove('dm_sb.txt')
#数据整理
for file in dir_list:
f=open(file)
file_name=re.match('.+_bz',file).group()
for line in f.readlines()[1:]:
fields=line.split(',')
if fields[3]=='0':
fields_sb=['','','','','']
fields_sb[0]=file_name
fields_sb[1]=re.sub('^010','',fields[0])
fields_sb[2]=fields[1]
fields_sb[3]=fields[5]
fields_sb[4]=fields[4]
fields_sb=','.join(fields_sb)+'\n'
f_sb.write(fields_sb)
else:
fields_cg=['','','','','']
fields_cg[0]=file_name
fields_cg[1]=re.sub('^010','',fields[0])
fields_cg[2]=fields[1]
fields_cg[3]=fields[3]
fields_cg[4]=fields[4]
fields_cg=','.join(fields_cg)+'\n'
f_cg.write(fields_cg)
f.close()
f_cg.close()
f_sb.close()