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

最新推荐文章于 2025-06-25 22:21:51 发布

原创最新推荐文章于 2025-06-25 22:21:51 发布 · 1.9k 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#python #csv #列表

Python 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种将两个CSV文件中的数据进行匹配和整合的方法，通过Python的csv模块，实现从ID.csv中查找蛋白质信息，并从name_or.csv中补充名称和生物体信息，最终将整合后的数据写入seq_sim_all.csv。

部署运行你感兴趣的模型镜像

需要将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])