Python-将2个文件与CSV模块和3列进行比较

Question

我正在尝试将2个文件与Python CSV模块进行比较。

第一个CSV文件如下所示：

Name N, Age, Pound
A, 1, 10
A, 2, 10
B, 2, 12

第二个CSV文件：

Name N, Age, Pound
A, 1, 10
A, 3, 8
C, 1, 7

如果列名和列龄相等，则在输出中返回两个文件的磅数。

输出文件示例：

Name, Age, Pound1, Pound2
A, 1, 10, 10

import csv
import io

alist, blist = [], []

with open("hosts.csv", "r") as fileA:
    reader = csv.reader(fileA, delimiter=',')
    for row in reader:
        alist.append(row)
with open("masterlist.csv", "r") as fileB:
    reader = csv.reader(fileB, delimiter=',')
    for row in reader:
        blist.append(row)

first_set = set(map(tuple, alist))
secnd_set = set(map(tuple, blist))

matches = set(first_set).intersection(secnd_set)
print (matches)

多谢了。

Answer 1

您可以尝试以下操作：

import csv
file1 = list(csv.reader(open('first.csv')))
file2 = list(csv.reader(open('second.csv')))
new_columns = [[a[0], a[1], a[-1], b[-1]] for a, b in zip(file1[1:], file2[1:]) if a[:2] == b[:2]]
write = csv.writer(open('outputfile.csv'))
write.writerow(new_columns)