从CSV文件中提取包含来自另一个CSV文件的任何值的行

Question

我一直在努力解决这个特定问题：

我有一个output.csv文件，看起来像这样：

另一个07.csv文件如下所示：

我想做的是从output.csv文件的mmsi列中获取值，并检查这些值中是否有07.csv中的任何行，然后将这些行提取到新的result072.csv文件中。 到目前为止，这是我的代码，但它做的很奇怪：它使这个新文件与07.csv一样大，并且包含值的行的外观更多。 我非常感谢您的反馈！ 先感谢您！

import csv
with open('output.csv') as hashes:
    hashes = csv.reader(hashes,delimiter = ',')
    hashes = set(col[1] for col in hashes)
    hashi= list(hashes)

with open('07.csv') as input_file, open('result072.csv', 'w') as output_file: 
    reader = csv.reader(input_file,delimiter = ',')
    writer = csv.writer(output_file,delimiter = ',')

for row in reader:
    if any(item in row for item in hashi):
        writer.writerow(row)

Answer 1

您需要做的是从output.csv文件中获取所有mmsi值，并逐步浏览07.​​csv，将每个条目与所有mmsi值进行比较。 如果匹配，则将该行写入新文件。

我对any功能都不满意，因此我手动进行了尝试。

import csv

with open('output.csv') as hashes:
    csv_data = csv.reader(hashes, delimiter = ',')
    mmsi_set = set(col[1] for col in csv_data)
    mmsi_list = list(hashes)


with open('07.csv') as input_file, open('result072.csv', 'w') as output_file: 
    reader = csv.reader(input_file, delimiter = ',')
    writer = csv.writer(output_file, delimiter = ',')

    for row in reader:
        for mmsi in mmsi_list:
            if mmsi == row[4]:
                writer.writerow(row)

上面的代码应将07.csv的任何行写入新文件result072.csv，其中output.csv的mmsi与07.csv的mmsi匹配