在Python中比较和附加.csv文件

Question

我有2.csv文件，我想比较和附加。假设列1文件1中有数字。我在第1列文件2中有一组相似的数字，但顺序不同，并不是所有的都出现了。 我想搜索文件1第1列，看看是否有匹配的数字在第1列文件2。如果有匹配项，我希望通过将列添加到匹配的编号，将文件2中的信息附加到文件1中。 我还删除了输入文件中的任何空行。在

例如：

File 1
Number Name     EmailAddress
1      Jenny    jj@ymail.com
2      Josh     jh@ymail.com
3      Tony     ty@gmail.org
4      Jeff     jf@youtube.cc


File2
Number Address  
4      123 Walnut St

1      17 Yerlington Wy
7      420 St B


Output File
Number Name  EmailAddress     Address
4      Jeff  jf@youtube.cc    123 Walnut St
1      Jenny jj@ymail.com     17 Yerlington Wy
7      NO ENTRY

这是我到目前为止得到的代码，它除了附加正确的信息之外，似乎什么都能做。在

^{pr2}$

Answer 1

首先，你的csv文件中没有分隔符，我不得不这样重写它们（逗号分隔符）：

在填充.csv公司名称：

Number,Name,EmailAddress
1,Jenny,jj@ymail.com
2,Josh,jh@ymail.com
3,Tony,ty@gmail.org
4,Jeff,jf@youtube.cc

infe2.csv：

^{pr2}$

那么你的代码就没有什么问题了：

• 您缺少空主机行的支票
• 奇怪的变量end_row正在做一些您不想做的事情，实际上是没有用的，因为您确实想追加master_row[1]
• not found的条件在错误的块中，因此多次追加“NO ENTRY”
• 存在未使用的变量row
• （只是增强）当文件的变量超出范围时，python将为您关闭文件，因此您可以删除close（）调用

在我的修复之后：

import csv

f1 = file('inFile.csv', 'rb')
f2 = file('inFile2.csv', 'rb')
f3 = file('outFile.csv', 'wb')

c1 = csv.reader(f1)
c2 = csv.reader(f2)
c3 = csv.writer(f3)

masterlist = [row for row in c2]
for hosts_row in c1:
    found = False
    for master_row in masterlist:
        results_row = hosts_row
        if hosts_row and master_row:
            if hosts_row[0] == master_row[0]:
                results_row.append(master_row[1])
                found = True
                break

    if not found:
        results_row.append('NO ENTRY')

    if any(hosts_row):
        c3.writerow(results_row)

输出文件：

Number,Name,EmailAddress,Address  
1,Jenny,jj@ymail.com,17 Yerlington Wy
2,Josh,jh@ymail.com,NO ENTRY
3,Tony,ty@gmail.org,NO ENTRY
4,Jeff,jf@youtube.cc,123 Walnut St

高温

Answer 2

另一个例子（这一个产生了你所展示的输出，但是我想Tomás的输出是你描述的）：

import csv

def read_file(path):
    with open(path) as fh:
        data = csv.reader(fh)
        return [row for row in data if row]

def match(l1, l2):
    result = []
    for i in l2:
        match = False
        for j in l1:
            if i[0] == j[0]:
                row = j + i[1:]
                match = True
                break
        if not match:
            row = [i[0], "NO ENTRY"]
        result.append(row)
    return result

if __name__ == '__main__':
    l1 = read_file("inFile.csv")
    l2 = read_file("inFile2")
    result = match(l1, l2)
    of = open('outFile.csv', 'w')
    writer = csv.writer(of)
    writer.writerows(result)
    of.close()

注意：

您应该使用open而不是file；在文件模式下，b用于二进制文件（通过在文本模式下打开可以进行行结束的转换）。还有很多，但已经解决了。在