字典KeyError

Question

我有以下代码：

def search_for_person(name):
    with open("address.txt", "r") as book:
        records = re.split("[-]+", book.read(), re.M)
        for data in records:
            record = get_record(data)
            if record['Name'] == name:
                print record


def get_record(string):
    return dict(re.findall("^(.*): (.*)$", string, re.M))

当我使用它来尝试定位一个记录时，我会得到以下输出：

Enter name: Daniel Ghi
{'Home Phone No.': 'Example', 'Mobile Phone No.': 'Example', 'Name': 'Daniel Ghi
', 'Address': 'Example'}
Traceback (most recent call last):
  File "address.py", line 35, in <module>
    search_for_person(name)
  File "address.py", line 18, in search_for_person
    if record['Name'] == name:
KeyError: 'Name'

正如您所看到的，这个确实打印出正确的数据，但是它仍然有一个KeyError，我不知道为什么。有人能指点我吗？

Answer 1

它打印出正确的数据，然后继续运行，并在以后的数据上引发KeyError。

很可能您的文件以一个------字符串结尾，所以您的split会生成一个在最末端带有一个空字符串的列表，然后生成一个空的dict，这个列表中自然没有任何键。

简单的修复方法是跳过data，如果它是空的，或者只有空格，或者record是空的。更健壮的修复方法是避免使用[]语法，当您无法确定每个dict实际拥有哪些键时；使用record.get('Name', None)代替。