python - 您可以使用 Python 来搜索 csv 中的值吗？

Question

我的目标是在一个小的 csv 文件(一列，几百行)中搜索一个值。 csv 中的行都是数字。这是我的尝试:

import csv

#read csv file
csv_reader = csv.reader(open('test.csv', 'rU'))

length_list = len(list(csv_reader))
value = []

for line in csv_reader:
    value.append(line) #transfer contents of csv to list

def find_voucher(input_voucher, counter):
    while counter < length_list:
        check_voucher = input_voucher in value[counter]
        if check_voucher == True:
            print "TRUE"
        else:
            print "FALSE"
    counter = counter + 1
find_voucher(1000,1)

当我运行脚本时出现此错误:

check_voucher = input_voucher in value[counter]
IndexError: list index out of range

Answer 1

csv.reader 是一个迭代器——在你遍历一次之后，它是空的，这与 list 不同。当你这样做时，你会用完它

length_list = len(list(csv_reader))

有很多方法可以简化你的程序，但如果你只是这样做

length_list = len(value)

在 for 循环之后而不是首先尝试找到长度，它应该可以工作。

如何简化程序的示例:

def find_voucher(input_voucher, filename):
    with open(filename, 'rU') as f:
        return ("%s\n" % input_voucher) in f

print find_voucher(1000, 'test.csv')