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


import globdef file_list():    files=[]    for file in glob.glob("*.pdb"):        files.append(file)    files=sorted(files,key=lambda x: int(x[:-4]))    return filesdef 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")

评论关闭