Python在2个csv文件中搜索

时间:2018-04-04 15:29:56

标签: python csv search

我有一个名为Callnum.csv的文件,如下所示:

SubjectID    Name      Code

66           Ioan     1101112
67           Attila   1134112
71           Lucian   1131312
72           Paul     1131212

和一个名为F-V-Mnum.csv的文件,看起来像这样

Group  PersonID    Code

1      78          1134112
2      18          1111112
1      27          1212131
3      59          1311110

我正在尝试编写一个程序,它解析第一个文件中的代码和第二个文件中的代码。如果这些行匹配,程序应该写入final.csv这样的内容:

SubjectID   Name     Code     Group   PersonID

67          Attila   1134112    1      78

我尝试了以下代码,但它不起作用:

import csv
import os

f = open('F-V-Mnum.csv', 'r')
reader = csv.reader(f)
next(reader)
f1 = open('Callnum.csv', 'r')
reader1 = csv.reader(f1, delimiter=',')
next(reader1)

ofile = open('final_file.csv', 'w')
writer = csv.writer(ofile)
for line1 in reader :
        for line2 in reader1 :
                if line1[2] == line2[2]:
                        writer.writerow(line2)

f.close()
f1.close()
ofile.close()

0 个答案:

没有答案