python按列合并数据_Python中基于匹配第一列的数据合并

我目前有两组数据文件，如下所示：

文件1：test1 ba ab cd dh gf

test2 fa ab cd dh gf

test3 rt ty er wq ee

test4 er rt sf sd sa

在文件2中：

^{pr2}$

我想根据第一列中的匹配行组合文件(以便“测试”匹配)

是这样的：test1 ba ab cd dh gf 123 344 123

test1 ba ab cd dh gf 234 567 787

test1 ba ab cd dh gf 221 344 566

test3 rt ty er wq ee 456 121 677

我有这个密码def combineFiles(file1,file2,outfile):

def read_file(file):

data = {}

for line in csv.reader(file):

data[line[0]] = line[1:]

return data

with open(file1, 'r') as f1, open(file2, 'r') as f2:

data1 = read_file(f1)

data2 = read_file(f2)

with open(outfile, 'w') as out:

wtr= csv.writer(out)

for key in data1.keys():

try:

wtr.writerow(((key), ','.join(data1[key]), ','.join(data2[key])))

except KeyError:

pass

但是输出结果如下：test1 ba ab cd dh gf 123 344 123

test3 er rt sf sd sa 456 121 677

有谁能帮我做输出，这样test1可以打印三次吗？在

非常感谢