用Python比较两个csv文件的ID，将查找到的ID信息提取出来

需要将ID.csv的蛋白质从name_or.csv中找到，并补充name和organism信息，最后写入了seq_sim_all.csv中

ID.csv

name_or.csv

seq_sim.all.csv

代码如下：

import csv
f = open('seq_sim_all.csv', 'w', encoding = 'utf-8', newline='')
csv_writer = csv.writer(f)
csv_writer.writerow(['Database ID', 'Name', 'Organism'])

with open('ID.csv','r',encoding='utf8') as fp1:
    # 使用列表推导式，将读取到的数据装进列表
    list1 = [i for i in csv.reader(fp1)]  # csv.reader 读取到的数据是list类型
    print(list1)
with open('name_or.csv','r',encoding='utf8') as fp2:
    list2 = [i for i in csv.reader(fp2)]
    print(list2)
for i in list1:
    for j in list2:
        if i[0] == j[0]:#匹配到ID，就将内容写入到seq_sim_all.csv中
            csv_writer.writerow([i[0], j[1], j[2]])
            print(i[0], j[0], j[1], j[2])