用pytho写的分割pdb文件一个脚本

import glob
def file_list():
    files=[]
    for file in glob.glob("*.pdb"):
        files.append(file)
    files=sorted(files,key=lambda x: int(x[:-4]))
    return files
def split_pdb(pdb_file,files_list,label1,label2):
    f_pdb=open(pdb_file,"r")
    lines=f_pdb.readlines()
    i=0;
    for line in lines:
        if i==12:
            break
        f_store=open(files_list[i],"a+")
        if label1 in line.strip():
            i+=1
            f_store.write(line)
            print "I am here"
            f_store.close()
            continue
        elif line.strip()==label2:
            continue
        else:
            f_store.write(line)
if __name__=="__main__":
    file_lists=file_list()
    split_pdb("hex6-dna.pdb1",file_lists,"TER","ENDMDL")

标签：

版权申明：本站文章部分自网络，如有侵权，请联系：west999com@outlook.com
特别注意：本站所有转载文章言论不代表本站观点！
本站所提供的图片等素材，版权归原作者所有，如需使用，请与原作者联系。