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

以下代码由PHP站长网 52php.cn收集自互联网

现在PHP站长网小编把它分享给大家，仅供参考

"""
python截取文件指定行进行合并.py
http://bbs.bathome.net/thread-38205-1-1.html
依山居 23:10 2015/11/24
懒得解释~
"""
import re
import os
import sys

datadirs="./DATAS/"
outputdirs="./DATAS/合并后/"

ls=os.listdir(datadirs)
rec=re.compile(r"([A-Z]{2}-[A-Z]d-[A-Z]d)-d{2}.csv")
fn={r for r in re.findall(rec,str(ls))}
fn=sorted(fn)

if len(fn)<1:
    input("没有找到文件")
    sys.exit(0)

for r in fn:
    csv1=datadirs+r+"-01.csv"
    csv2=datadirs+r+"-02.csv"
    csv3=datadirs+r+"-03.csv"
    out=r+".csv"

    if os.path.exists(csv1):
        with open(csv1) as f:
            c1=f.readlines()[:3]
    if os.path.exists(csv2):
        with open(csv2) as f:
            c2=f.readlines()[3:]
        c1=c1+c2
    if os.path.exists(csv3):
        with open(csv3) as f:
            c3=f.readlines()[3:]
        c1=c1+c3
    with open(outputdirs+out,"w+") as f:
        f.writelines(c1)

以上内容由PHP站长网【52php.cn】收集整理供大家参考研究

如果以上内容对您有帮助，欢迎收藏、点赞、推荐、分享。

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!