如何根據兩個文件中的公共信息合併兩個CSV文件？

Question

Evans， 非常感謝。這幾乎是預期的結果。請按照以下步驟進行修改。 我們需要一點改變。請看看圖像。 在當前結果中，fileOne中的每條記錄搜索fileTwo中類似的adv_id和user_id，並在查找記錄時查找並停止。但可能性是fileTwo中可能有幾個類似的記錄。所以，我們需要fileTwo中的所有類似記錄。 fileOne的所有記錄必須至少在fileTwo中可用一次或多次。因此，我們應該包含fileOne的所有記錄以及來自fileTwo的所有類似記錄。我認爲逐行搜索可能會有所幫助。這是fileOne的第一個文件的adv_id和user_id，並搜索fileTwo中的所有記錄以查找相似的記錄。接下來使用fileOne的第2條記錄並搜索fileTwo中的所有記錄。等等。

Revised Image For Expected Result

Answer 1

下面的腳本將根據您的原始樣本數據創建result.csv（見過去的編輯質疑）：

import csv 
from collections import defaultdict 

d_entries = defaultdict(list) 

with open('fileTwo.csv', 'r') as f_fileTwo: 
    csv_fileTwo = csv.reader(f_fileTwo) 
    header_fileTwo = next(csv_fileTwo) 
    for cols in csv_fileTwo: 
     d_entries[(cols[0], cols[1])].append([cols[0], ''] + cols[1:]) 

with open('fileOne.csv', 'r') as f_fileOne, open('result.csv', 'w', newline='') as f_result: 
    csv_fileOne = csv.reader(f_fileOne) 
    csv_result = csv.writer(f_result) 
    header_fileOne = next(csv_fileOne) 
    csv_result.writerow(header_fileOne) 

    for cols in csv_fileOne: 
     if (cols[0], cols[2]) in d_entries: 
      csv_result.writerow(cols) 
      csv_result.writerows(d_entries.pop((cols[0], cols[2]))) 

result.csv然後將含有當在Excel中打開以下數據：

使用Python測試3.4.3

要只在adv_id欄比賽，並有所有條目：

import csv 
from collections import defaultdict 

d_entries = defaultdict(list) 

with open('fileTwo.csv', 'r') as f_fileTwo: 
    csv_fileTwo = csv.reader(f_fileTwo) 
    header_fileTwo = next(csv_fileTwo) 
    for cols in csv_fileTwo: 
     d_entries[cols[0]].append([cols[0], ''] + cols[1:]) 

with open('fileOne.csv', 'r') as f_fileOne, open('result.csv', 'w', newline='') as f_result: 
    csv_fileOne = csv.reader(f_fileOne) 
    csv_result = csv.writer(f_result) 
    header_fileOne = next(csv_fileOne) 
    csv_result.writerow(header_fileOne) 

    for cols in csv_fileOne: 
     if cols[0] in d_entries: 
      csv_result.writerows(d_entries.pop(cols[0])) 
     csv_result.writerow(cols)