如何在csv文件中搜索公共元素并保存整行

时间:2015-07-03 18:26:49

标签: python csv

我对python来说是全新的......

我需要搜索两个csv文件(Workbook1和Workbook2)的列,无论它们是否共享。然后我想将整行存储到一个文件中,可能是一个csv。以下是我被阻止的地方:

stack = []
import csv
with open('/Users/freemanlab/Downloads/Workbook1.csv', newline= '') as f:
    reader = csv.reader(f, delimiter=':', quoting=csv.QUOTE_NONE)
    for row in reader:
        stack.append(row)

list1 = stack

stack = []
import csv
with open('/Users/freemanlab/Downloads/Workbook2.csv', newline= '') as f:
    reader = csv.reader(f, delimiter=':', quoting=csv.QUOTE_NONE)
    for row in reader:
        stack.append(row)

list2 = stack

matches = [item for item in list1 if item in list2]

print(matches)
print('Number of matches is...')
print(len(matches))

with open('/Users/freemanlab/Downloads/matches.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerows(matches)

我知道这样,它只适用于只有一列的csv的第一列,但是我想在更复杂的文件上工作,例如,在文件的第二列中有更多的文件

最后,我想在第三个文件(Workbook3)中搜索是否也找到了常见内容,当它找到它时,将整行存储到csv中。

非常感谢你的帮助。

0 个答案:

没有答案