如何使用python增強我的數據集輸出

Question

我有2個輸入文件，input.txt和datainput.txt。我檢查input.txt的第二列是否與datainput.txt的第一列匹配，如果它們匹配，那麼我把它的orthodb_id放在輸出文件的結尾相關行。

input.txt中：

5 21 218 
6 11 1931 
7 26 173 

datainput.txt：

>21|95|28|5 
Computer 
>11|28|5|5 
Cate 

code.py：

import csv 

with open('input.txt', 'rb') as file1: 
    file1_data = dict(line.split(None, 2)[1::-1] for line in file1 if line.strip()) 

with open('data.txt', 'rb') as file2, open('output.txt', 'wb') as outputfile: 
    output = csv.writer(outputfile, delimiter='|') 
    for line in file2: 
     if line[:1] == '>': 
      row = line.strip().split('|') 
      key = row[0][1:] 
      if key in file1_data: 
       output.writerow(row + [file1_data[key]]) 

這是輸出我用我的代碼獲得：

>21|95|28|5|5 
>11|28|5|5|6 

Answer 1

你只需要在代碼中添加一個else塊，以獲得所需的輸出：

import csv 

with open('input.txt', 'rb') as file1: 
    file1_data = dict(line.split(None, 2)[1::-1] for line in file1 if line.strip()) 

with open('data.txt', 'rb') as file2, open('output.txt', 'wb') as outputfile: 
    output = csv.writer(outputfile, delimiter='|') 
    for line in file2: 
     if line[:1] == '>': 
      row = line.strip().split('|') 
      key = row[0][1:] 
      if key in file1_data: 
       output.writerow(row + [file1_data[key]]) 
     else: 
      outputfile.write(line)