记录人生中第一个python脚本

功能： 将A_pep.fa里的“>”开头行改为“>sp._id@sequence_id”

#!/usr/bin/python
# -*- coding: UTF-8 -*-

list=['A','B','C','D','E','F']
for name in list:
    filename = name+"_pep.fa"
    outname = name+"_pep_shroten.fa"
    infile = open(filename,"r")
    outfile = open(outname,"w")
    for line in infile:
            if line[0] == ">":
                newid = (line.split(" ")[0]).split(">")[1]
                outfile.write(">"+name+"@"+newid)
            else: outfile.write(line)
    infile.close()
    outfile.close()