python 文件合并和编号

# -*- coding:utf-8 -*-
import os
import re
p1=r"([0-9][0-9][AB]).w{3}$"
pattern1=re.compile(p1)
def get_dir(zz):
    listdir=[]
    for filename in os.listdir('./%s'%zz):
        listdir.append(filename)
        # print(filename)
        # print(type(filename))
    # print(listdir)
    return listdir

def joint_b0(listdir,zz):
    if not os.path.exists('./new/'):
        os.makedirs('./new/')
    fw=open('./new/%s.csv'%(zz),'a')
    for i in listdir:
        # print(i)
        j=1000
        matcher1=re.findall(pattern1,i)
        fr=open('./%s/%s'%(zz,i))
        for line in fr.readlines():
            try:
                # print(len(line))
                if(len(line)==1):
                    continue
                else:
                    fw.write(matcher1[0]+'_%s'%j+','+line.strip('
')+'
')
                    j=j+1
            except:
                pass
    fr.close()
    fw.close()    

def joint_b1(listdir,zz):#单独拼接
    if not os.path.exists('./new/'):
        os.makedirs('./new/')
    fw=open('./new/%s.csv'%(zz),'a')
    for i in listdir:
        # print(i)
        j=1
        matcher1=re.findall(pattern1,i)
        fr=open('./%s/%s'%(zz,i))
        for line in fr.readlines():
                fw.write(line)

    fr.close()
    fw.close()    
if __name__=="__main__":

    # dir_list=['b0'],'b3','b4','b2_idsd','b3_idcd'


    dir_list=['b4']#用于编号和拼接，会在new目录下生成编号号码的文件
    for i in range(len(dir_list)):
        listdir=get_dir(dir_list[i])
        # print(listdir)
        joint_b0(listdir,dir_list[i])

    # dir_list=['stock']#用于拼接，将需要拼接的放到stock目录下，会在new目录下生成stock文件，然后根据需要修改名称
    # for i in range(len(dir_list)):
    #     listdir=get_dir(dir_list[i])
    #     # print(listdir)
    #     joint_b1(listdir,dir_list[i])