Python：比较2个CSV，如果某个单元格匹配，就把数据拉过来

Question

文件A:

id,desc,name
12345,blah blah blah,jsmith
6789,yada yada yada,ckast
54321,yum yum yum,jpetersen

文件B：

key,id
AB-873,6789
CF-395,54321
HG-713,12345

我想做的是查看文件A中的每一行，看看id列是否与文件B中的id列匹配，以及它是否通过“name”单元格复制到文件B中。因此，文件B最终将如下所示：

AB-873,6789,ckast
CF-395,54321,jpetersen
HG-713,12345,jsmith

我知道“csv”Python模块可以读取单独的行，但是我不知道从哪里开始。谢谢！你知道吗

Answer 1

如果您想要一个简单的代码，此代码适用于您：

a_lines = open('FileA', 'r').readlines()[1:]
b_lines = open('FileB', 'r').readlines()[1:]
file_result = open('result', 'w')

# Read content of FileA to a table (list of lists)
a_table = []
for l in a_lines:
    a_table.append([w.strip() for w in l.split(',')])

# Read content of FileB in a dictionary.
# The 'id' field as dictionary key for simple look-up.
b_dict = {}
for l in b_lines:
    words = l.split(',')
    b_dict[words[1].strip()] = words[0].strip()

# Do the actual work and save result.
for row in a_table:
    if row[0] in b_dict:
        file_result.write(b_dict[row[0]] + ',' + row[0] + ',')
        file_result.write(row[2] + '\n')

我用你的样品测试了一下。你知道吗