比较两个CSV文件并搜索相似的项目 [英] Compare two CSV files and search for similar items

问题描述

因此，我有两个CSV文件，它们试图进行比较并获得相似项目的结果. 第一个文件hosts.csv如下所示:

So I've got two CSV files that I'm trying to compare and get the results of the similar items. The first file, hosts.csv is shown below:

Path    Filename    Size    Signature
C:\     a.txt       14kb    012345
D:\     b.txt       99kb    678910
C:\     c.txt       44kb    111213

第二个文件masterlist.csv如下所示:

The second file, masterlist.csv is shown below:

Filename    Signature
b.txt       678910
x.txt       111213
b.txt       777777
c.txt       999999

您可以看到行不匹配，并且masterlist.csv始终大于hosts.csv文件.我要搜索的唯一部分是签名部分.我知道这看起来像:

As you can see the rows do not match up and the masterlist.csv is always larger than the hosts.csv file. The only portion that I'd like to search for is the Signature portion. I know this would look something like:

hosts[3] == masterlist[1]

我正在寻找一种解决方案，该解决方案将为我提供以下内容(基本上是带有新的RESULTS列的hosts.csv文件):

I am looking for a solution that will give me something like the following (basically the hosts.csv file with a new RESULTS column):

Path    Filename    Size    Signature    RESULTS
C:\     a.txt       14kb    012345       NOT FOUND in masterlist
D:\     b.txt       99kb    678910       FOUND in masterlist (row 1)
C:\     c.txt       44kb    111213       FOUND in masterlist (row 2)

我搜索了帖子，发现与

I've searched the posts and found something similar to this here but I don't quite understand it as I'm still learning python.

编辑使用Python 2.6

Edit Using Python 2.6

推荐答案

当我的解决方案正常运行时，请查看下面的Martijn答案，以寻求更有效的解决方案.

While my solution works correctly, check out Martijn's answer below for a more efficient solution.

您可以在此处中找到python CSV模块的文档.

You can find the documentation for the python CSV module here.

您正在寻找的东西是这样的:

What you're looking for is something like this:

import csv

f1 = file('hosts.csv', 'r')
f2 = file('masterlist.csv', 'r')
f3 = file('results.csv', 'w')

c1 = csv.reader(f1)
c2 = csv.reader(f2)
c3 = csv.writer(f3)

masterlist = list(c2)

for hosts_row in c1:
    row = 1
    found = False
    for master_row in masterlist:
        results_row = hosts_row
        if hosts_row[3] == master_row[1]:
            results_row.append('FOUND in master list (row ' + str(row) + ')')
            found = True
            break
        row = row + 1
    if not found:
        results_row.append('NOT FOUND in master list')
    c3.writerow(results_row)

f1.close()
f2.close()
f3.close()

这篇关于比较两个CSV文件并搜索相似的项目的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！