搜索CSV文件，如果它不在文件中

Question

我有一个装满票证信息的CSV文件。我创建了一个小脚本，将票证编号以空格分隔输入到列表中，该列表搜索CSV文件中的每个票证，如果找到票证，它会输出该行的信息。

我的问题是，如果我搜索的票证不在CSV文件中，它只会跳过它并继续，但我希望它告诉我该票证不在文件中。据我所知，它正在按行或行搜索CSV文件。如果我尝试一个else语句，它将开始打印CSV文件中的每一行，如果该票据不在该行中。

我需要能够输入多个票号并让python单独搜索每个票号。如果它在第1列中找到票证，则从该行打印信息，如果在第1列的任何行中没有找到票证，则打印出该票证，后跟“不在文件中”。

import csv

file = csv.reader(open('Path To CSV file', "rb"), delimiter=",")
newticket = raw_input('Enter Ticket Numbers:').split()

for line in file:
    for tickets in newticket:
        if tickets == line[1]:
            print(tickets, line[36], line[37])

Answer 1

如果您的CSV文件不是很大，我建议您阅读整个内容，将您想要的数据提取到字典中，并以票号为键。搜索dict非常快。

我修改了您的文件打开代码，以使用with语句，如果读取文件时出现问题，该语句将正常失败。顺便说一句，没有必要将逗号指定为CSV分隔符，因为这是默认的分隔符。

import csv

with open('Path To CSV file', "rb") as f:
    data = {line[1]: (line[36], line[37]) for line in csv.reader(f)}

newtickets = raw_input('Enter Ticket Numbers:').split()

for ticket in newtickets:
    line = data.get(ticket)
    if line:
        print(ticket, line)
    else:
        print(ticket, "not found")

dict.get方法返回无，如果键不在判决中，尽管如果需要，您可以指定另一个默认返回值。所以我们可以像这样重写该部分：

for ticket in newtickets:
    line = data.get(ticket, "not found")
    print(ticket, line)

或者，我们可以在操作符中使用：

for ticket in newtickets:
    if ticket in data:
        print(ticket, data[ticket])
    else:
        print(ticket, "not found")

所有3个版本的效率大致相同，请选择您认为最具可读性的版本。