python截取文件指定行进行合并.py

依山居

1. """
   
2. python截取文件指定行进行合并.py
   
3. http://bbs.bathome.net/thread-38205-1-1.html
   
4. 依山居 23:10 2015/11/24
   
5. 懒得解释~
   
6. """
   
7. import re
   
8. import os
   
9. import sys
   
10. 
    
11. datadirs="./DATAS/"
    
12. outputdirs="./DATAS/合并后/"
    
13. 
    
14. ls=os.listdir(datadirs)
    
15. rec=re.compile(r"([A-Z]{2}-[A-Z]\d-[A-Z]\d)-\d{2}.csv")
    
16. fn={r for r in re.findall(rec,str(ls))}
    
17. fn=sorted(fn)
    
18. 
    
19. if len(fn)<1:
    
20.     input("没有找到文件")
    
21.     sys.exit(0)
    
22. 
    
23. for r in fn:
    
24.     csv1=datadirs+r+"-01.csv"
    
25.     csv2=datadirs+r+"-02.csv"
    
26.     csv3=datadirs+r+"-03.csv"
    
27.     out=r+".csv"
    
28. 
    
29.     if os.path.exists(csv1):
    
30.         with open(csv1) as f:
    
31.             c1=f.readlines()[:3]
    
32.     if os.path.exists(csv2):
    
33.         with open(csv2) as f:
    
34.             c2=f.readlines()[3:]
    
35.         c1=c1+c2
    
36.     if os.path.exists(csv3):
    
37.         with open(csv3) as f:
    
38.             c3=f.readlines()[3:]
    
39.         c1=c1+c3
    
40.     with open(outputdirs+out,"w+") as f:
    
41.         f.writelines(c1)
    
42. 
    
43.         
    
44.         
    
45.         
    
46.             
    
47.         
    

复制代码