如何检查输入是否超出词典列表的范围?

2024-05-16 19:56:23 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个叫做“周期表”的文件。在这个文件中有多行。每一行的侧面都有一个原子序数,右边有一个相应的元素名,如下所示:

1 Hydrogen
2 Helium
3 Lithium
4 Beryllium
5 Boron
6 Carbon
7 Nitrogen
8 Oxygen
9 Fluorine
10 Neon
11 Sodium
12 Magnesium
13 Aluminium
14 Silicon
etc...

我编写了一个程序,要求输入元素名称或数字,并在字典中打印出相应的值。如果用户输入1,它将打印氢气,同样,如果用户输入硅,它将输出14。但是,我希望程序在用户输入不存在的原子序数(如150)或不存在的元素(如Blanket或任何其他字符串)时通知用户。我尝试使用if,但它打印出一个无限循环:

^{pr2}$

Tags: 文件用户程序元素carbon周期表heliumlithium
3条回答

为了有效地解决这个问题(无需反复读取文件),您需要将其分解为两个步骤,您当前的代码将这两个步骤混合在一起。首先,读取文件并准备元素名和数字之间的字典映射。第二,处理用户输入并对照字典进行检查。在

# step 1, build the mapping
element_list = {}
with open("periodic_table.txt") as f:
    for line in f:
        number, name = line.split()
        element_list[number] = name
        element_list[name] = number # map in both directions

# step 2, test user input (quits after an empty input)
user_entry = input("Enter element number or name: ")
while user_entry:
    try:
        print(element_list[user_entry])
    except KeyError:
        print("Unrecognized number or name.")
    user_entry = input("Enter element number or name: ")

对于对现有代码的最小更改,如果找到元素,可以设置一个标志found,并对其进行相应的操作。所以:

    found = False
    for line in f:
        # ....
        if name == key:
            print(val)
            found = True
        elif ...
    if not found:
        print("Not an element or atomic number: {}".format(name))

一旦找到匹配项,您可以直接退出程序。如果没有找到匹配项,将再次提示用户输入。否则,程序将在打印相应的编号/名称后终止。在

# http://docs.python.org/2/library/sys.html
import sys

#variable below is not doing much!
#element_list = {}
name = input("Enter element number or element name: ")
#Changed loop to be infinite
while True:
    with open("periodic_table.txt") as f:
        for line in f:
           (key, val) = line.split()
           element_list[int(key)] = val
           if name == key:
               print(val)
               sys.exit()
           elif name == val:
               print(key)
               sys.exit()
    #If something is found, it will never reach this
    print("No match found... try again!")
    name = input("Enter element number or element name: ")

相关问题 更多 >