RSV——根据id合并文件内容

  1 from Bio import SeqIO
  2 import pandas as pd
  3 
  4 def extract_ids(fasta_file):
  5     ids = []
  6     with open(fasta_file, "r") as handle:
  7         for record in SeqIO.parse(handle, "fasta"):
  8             ids.append(record.id)
  9 
 10     return ids
 11 
 12 def read_csv(csv_file):
 13     df = pd.read_csv(csv_file)
 14     return df
 15 
 16 
 17 def main():
 18     fasta_file = 'em59.fa'
 19     ids = extract_ids(fasta_file)
 20     print(len(ids))
 21     csv_file = 'result.csv'
 22     df = read_csv(csv_file)
 23     print(len(df.index))
 24 
 25     #df.columns = ids
 26     df.index = ids
 27     df.to_csv('json_plddt.csv')
 28     print(df)
 29 
 30 
 31 if __name__ == '__main__':
 32     main()