使用用户输入搜索文件

Question

我在尝试使用用户输入从文本文件中打印出某些元素时遇到了一些困难。

我有一个文本文件，其中分别显示 ID 号、目的地名称、票价和可用座位数，如下所示：

• Z5, 墨尔本, 150, 30
• Z6, 纽约, 120, 60
• Z9, 首尔, 100, 50
• Z2, 东京, 170, 40

我想要用户输入 ID 号或目的地名称，程序将显示 (1) 对应的目的地（如果使用了 ID 号）或重复 (2 ) 目的地名称（如果使用了目的地名称）。

也就是说，用户输入 Z6，程序将显示 New York，或者如果用户输入 New York，则将再次打印 New York。

下面的代码可以打印出ID号对应的目的地；但是，如果用户输入目的地的名称，我似乎无法打印出目的地的名称，因为我得到一个列表索引超出范围错误。

search = str (input ('Please enter the name or ID number of the destination:\n'))  
info = open ('destinations.txt', 'r').readlines()
for I in info:
    data = I.split (',')
    if search == data[0]:
     print (data[1])
    elif search == data[1]:
     print ([data[1])

Answer 1

我复制了您的代码，唯一需要修复的是最后一行的语法错误：print([data[1]) 在data 之前有一个额外的开放方括号。以下代码正常工作：

search = str (input ('Please enter the name or ID number of the destination:\n'))  
info = open ('destinations.txt', 'r').readlines()
for I in info:
    data = I.split (',')
    if search == data[0]:
        print (data[1])
    elif search == data[1]:
        print (data[1])

您还可以使用or 语句将这两个条件合二为一：

if search == data[0] or search == data[1]:
    print(data[1])

确保您的 destinations.txt 看起来像这样：

Z5,Melbourne,150,30
Z6,New York,120,60
Z9,Seoul,100,50
Z2,Tokyo,170,40

因为这是您的代码所期望的格式。