python - 如何检查列表中的项目是否出现在 csv 中？

Question

我有一个项目列表:

List = ['apple', 'banana', 'orange']

和一个csv文件:

apple
kiwi
banana
orange
pear

我试图查看列表中的每个项目是否与 csv 的第一列匹配，如果匹配，则在行末尾附加一个“1”(否则附加一个“0”)，但是我有点迷路了。下面的代码几乎就是我想要得到的最终结果。

apple, 1
kiwi, 0
banana, 1
orange, 1
pear, 0

我非常是 python 的新手，非常感谢任何帮助!谢谢

Answer 1

我试过下面的代码，它将结果存储在 CSV 中

import csv

lists = ['apple', 'banana', 'orange']
nf = csv.writer(open("output.csv", "w"))
with open('input.csv', 'rb') as f:
    reader = csv.reader(f)
    for row in reader:
        if(''.join(row) in lists):
            nf.writerow([''.join(row),1])
        else:
            nf.writerow([''.join(row),0])

CSV 格式的输出

apple,1
kiwi,0
banana,1
orange,1
pear,0