Python:比较两个csv文件中的特定列

Question

假设我有两个CSV文件(file1和file2),其内容如下所示:

文件1:

fred,43,Male,"23,45",blue,"1, bedrock avenue"

文件2:

fred,39,Male,"23,45",blue,"1, bedrock avenue"

我想比较这两个CSV记录,看看列0,2,3,4和5是否相同.我不关心第1列.

这样做的最pythonic方法是什么？

编辑:

一些示例代码将不胜感激.

EDIT2:

请注意,嵌入的逗号需要正确处理.

Answer 1

我想最好的方法是使用Python库:http://docs.python.org/library/csv.html.

更新(示例已添加):

import csv
reader1 = csv.reader(open('data1.csv', 'rb'), delimiter=',', quotechar='"'))
row1 = reader1.next()
reader2 = csv.reader(open('data2.csv', 'rb'), delimiter=',', quotechar='"'))
row2 = reader2.next()
if (row1[0] == row2[0]) and (row1[2:] == row2[2:]):
    print "eq"
else:
    print "different"

Answer 2

>>> import csv
>>> csv1 = csv.reader(open("file1.csv", "r"))
>>> csv2 = csv.reader(open("file2.csv", "r"))
>>> while True:
...   try:
...     line1 = csv1.next()
...     line2 = csv2.next()
...     equal = (line1[0]==line2[0] and line1[2]==line2[2] and line1[3]==line2[3] and line1[4]==line2[4] and line1[5]==line2[5])
...     print equal
...   except StopIteration:
...     break
True

更新

3年后，我想我宁愿这样写。

import csv

interesting_cols = [0, 2, 3, 4, 5]

with open("file1.csv", 'r') as file1,\
     open("file2.csv", 'r') as file2:

    reader1, reader2 = csv.reader(file1), csv.reader(file2)

    for line1, line2 in zip(reader1, reader2):
        equal = all(x == y
            for n, (x, y) in enumerate(zip(line1, line2))
            if n in interesting_cols
        )
        print(equal)