在Python中，从文本文件的列中删除前导零，然后将两个文本文件中具有相同字符串的类似行合并为一个

Question

假设我有2个文件-

File 1:

00023 apple
00632 banana
00020 cat
00290 dog
00004 egg
00089 hen
00310 ink

File 2:

45 fan
23 jar
339 goat
4 kite
310 land

因此，首先在文件1中，我需要修剪第1列的前导零。我们将得到如下结果：

23 apple
632 banana
20 cat
290 dog
4 egg
89 hen
310 ink

在此之后，我想比较文件1和文件2中的公共编号（在本例中为23、4、310），然后将公共列合并到一个新文件中，并生成类似以下内容的输出-

23 apple jar
4 egg kite
310 ink land

我已经在这里找到了一些答案，但无法真正理解，因为我对这一点还不熟悉

Answer 1

这可以使用标准库中的csv模块完成

首先，使用字典理解将每个文件读入字典（可以使用str.lstrip方法删除前导零）

接下来，利用dictionary键视图支持intersection等集合操作这一事实来查找公共键

最后，将公共密钥和关联值写入新的csv文件

import csv 


with open('csv1.csv', newline='') as f:
    reader = csv.reader(f)
    data1 = {col1.lstrip('0'): col2 for col1, col2 in reader}


with open('csv2.csv', newline='') as f:
    reader = csv.reader(f)
    data2 = {col1: col2 for col1, col2 in reader}


common_keys = data1.keys() & data2.keys()


with open('merge.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    for key in common_keys:
        writer.writerow([key, data1[key], data2[key]])