按公用列组合2个.csv文件

Question

我有两个.csv文件，文件1的第一行是：

MPID,Title,Description,Model,Category ID,Category Description,Subcategory ID,Subcategory Description,Manufacturer ID,Manufacturer Description,URL,Manufacturer (Brand) URL,Image URL,AR Price,Price,Ship Price,Stock,Condition

文件2的第一行：

Regular Price,Sale Price,Manufacturer Name,Model Number,Retailer Category,Buy URL,Product Name,Availability,Shipping Cost,Condition,MPID,Image URL,UPC,Description

然后每个文件的其余部分都充满了信息。

如您所见，两个文件都有一个名为MPID的公共字段（文件1:col 1，文件2:col 9，其中第一个col是col1）。

我想创建一个新文件，通过查看此列将这两个文件组合在一起（如中所示：如果两个文件中都有MPID，则在新文件中，该MPID将显示为来自文件1的行和来自文件2的行）。如果一个MPID只出现在一个文件中，那么它也应该进入这个组合文件。

文件没有以任何方式排序。

如何在一个带有shell脚本或python的debian机器上执行此操作？

谢谢。

编辑：除了分隔字段的文件外，两个文件都没有逗号。

Answer 1

这就是经典的“关系连接”问题。

你有几个算法。

• 嵌套循环。你从一个文件中读取来选择一个“主”记录。您将读取整个其他文件，查找与主文件匹配的所有“详细信息”记录。这是个坏主意。

• 排序合并。根据公用密钥将每个文件排序为临时副本。然后，通过读取主控文件，然后从详细信息中读取所有匹配的行并写入合并的记录来合并这两个文件。

• 查找。您将其中一个文件完全读入内存中的字典中，并由关键字字段索引。对于detail文件来说，这可能很棘手，在该文件中，每个键都有多个子项。然后读取另一个文件并在字典中查找匹配的记录。

其中，排序合并通常是最快的。这完全是使用unix sort命令完成的。

查找实现

import csv
import collections

index = collections.defaultdict(list)

file1= open( "someFile", "rb" )
rdr= csv.DictReader( file1 )
for row in rdr:
    index[row['MPID']].append( row )
file1.close()

file2= open( "anotherFile", "rb" )
rdr= csv.DictReader( file2 )
for row in rdr:
    print row, index[row['MPID']]
file2.close()