python - 如果行 [0] 在行 [1] 中打印行

Question

我有一个包含 2 列的 csv 文件。我只是想弄清楚每个 row[0] 值是否在某个 row[1] 中，如果是，则打印 row。

csv 文件中的项目:

COL1,   COL2
1-A,    1-A
1-B,    2-A
2-A,    1-B
2565,   2565
51Bc,   51Bc
5161,   56
811,    65
681,    11
55,     3
3,      55

代码:

import csv
doc= csv.reader(open('file.csv','rb'))

for row in doc:
    if row[0] in row[1]:
        print row[0]

最终结果应该是:

1-A
1-B
2-A
2565
51Bc
55
3

相反，它给了我:

1-A
2565
51Bc

它打印这些数字，因为它们并排相邻，但我需要它做的是获取 COL1 中的第一项，看看是否在整个 COL2 列表中找到它，如果找到则打印。不看是否相邻并打印它。

Answer 1

当您说for row in doc时，它仅获取一对元素并将它们放入row中。因此，row[1] 不可能在任何时间点保存整个列。您需要执行初始循环以将该列作为列表获取，然后再次循环 csv 文件以进行比较。实际上，您可以将两列存储在单独的列表中，并且只需打开文件一次。

import csv
doc= csv.reader(open('file.csv','rb'))

# Build the lists.
first_col = []
second_col = set()
for row in doc:
    first_col.append(row[0])
    second_col.add(row[1])

# Now actually do the comparison.
for item in first_col:
    if item in second_col:
        print item

根据 abarnert 的建议，我们对第二列使用 set()。 set 已针对查找其中的值进行了优化，这就是我们用它所做的一切。 list 针对循环每个元素进行了优化，这就是我们对 first_col 所做的事情，因此这更有意义。