Python比较lin中分隔字符的文件

Question

在那里。 我是python中的乞丐，我正在努力做到以下几点：

我有这样一个文件（+10k线）：

EgrG_000095700 /product="ubiquitin carboxyl terminal hydrolase 5"
EgrG_000095800 /product="DNA polymerase epsilon subunit 3"
EgrG_000095850 /product="crossover junction endonuclease EME1"
EgrG_000095900 /product="lysine specific histone demethylase 1A"
EgrG_000096000 /product="charged multivesicular body protein 6"
EgrG_000096100 /product="NADH ubiquinone oxidoreductase subunit 10"

这个（+600行）：

EgrG_000076200.1
EgrG_000131300.1
EgrG_000524000.1
EgrG_000733100.1
EgrG_000781600.1
EgrG_000094950.1

第二个文件的所有ID都在第一个文件中，因此我希望第一个文件的行与第二个文件的ID对应。你知道吗

我写了以下脚本：

f1 = open('egranulosus_v3_2014_05_27.tsv').readlines()
f2 = open('eg_es_final_ids').readlines()
fr = open('res.tsv','w')

for line in f1:
     if line[0:14] == f2[0:14]:
        fr.write('%s'%(line))

fr.close()
print "Done!"

我的想法是搜索分隔每行字符的id，以便将一个文件的EgrG\u XXXX与另一个文件匹配，然后将这些行写入一个新文件。 我尝试了一些修改，那只是我想法的“核心”。 我什么都没有。在其中一次修改中，我只得到了一行。你知道吗

Answer 1

我将把来自f2的id存储在一个集合中，然后检查f1。你知道吗

id_set = set()
with open('eg_es_final_ids') as f2:
    for line in f2:
        id_set.add(line[:-2]) #get rid of the .1

with open('egranulosus_v3_2014_05_27.tsv') as f1:
    with open('res.tsv', 'w') as fr:
        for line in f1:
            if line[:14] in id_set:
                fr.write(line)