Question

我正在写＆＃34; TER＆＃34;在特定残留数之后的outfile中的一行之后（与参考文件中的残留数匹配）。代码不会给我任何错误，但只有来自infile的副本并没有添加单词。当我尝试在outfile中搜索时尝试打印一些代码时，我认为它会中断。

这就是一部分。另一个是我如何确保＆＃34; TER＆＃34;仅在最后一个残留数字之后添加（它们在彼此之下的行中有各种数量，我只想在最后一个之后添加它们）。这是我的代码：

import sys
import argparse

def main(argv):
    parser = argparse.ArgumentParser(description='Read SSBOND directives from a PDB, and generate corresponding CONECT records')
    parser.add_argument('infile', help='input file (PDB format)')
    parser.add_argument('outfile', help='output file (PDB format)')
    parser.add_argument('reference', help =' ref')
    args = parser.parse_args()

    resnum_1 =[]
    res = []

    with open(args.infile, "r") as f, open(args.outfile, "w+") as of, open(args.reference,"r") as rf:
        for line in rf:
            if line[0:4] == "TER ":
                resnum = line[22:27]
                resname = line[17:20]
                chain = line[21]
                resnum_1.append(resnum)

                for line in f:
                    of.write(line)

            for line in of:
                if line[0:6] == "ATOM  ":
                    resnum_fo = line[22:27]
                    resname_fo = line[17:19]
                    chain_fo = line[21]
                    res.append(resnum_fo)


                    if resnum in resnum_1 and resnum_fo in res:
                        of.write("TER\n")

if __name__ == "__main__":
    main(sys.argv)

非常感谢！

文件如下所示：

ATOM      0  HB2 CYX D 452      45.962  -2.641 -17.575  1.00  0.00  
ATOM      0  HB3 CYX D 452      46.188  -2.186 -19.050  1.00  0.00  
TER     995      CYX D 452  
ATOM    995  N   ARG D 492      42.476  10.547 -39.562  1.00  0.00

Answer 1

几点/猜测：

您以of的身份打开w+，然后执行for line in of:。你是说f吗？（我会建议一些更具描述性的名字）
您似乎检查"ATOM "（注意 2 空格）。从您的文件看起来它应该只有1个空格。

为了解决空间问题，我建议您使用split：

 "one two three".split()
 ['one', 'two', 'three']

因此，对于您的情况，请使用：

for line in rf:
    contents = line.split()
    if contents[0] == "TER":
        ...


for line in f:
    contents = line.split()
    if contents[0] == "ATOM":
        ...

在特定行Python之后添加一些东西

1 个答案: