Python字典值更新问题：update_availability函数KeyError报错排查

需求回顾 #

实现update_availability函数，接收三个参数：图书馆目录（字典类型）、书名、新的可用性状态，更新目录中对应书籍的可用性。

现有代码问题 #

原代码在书名不存在于目录时，直接尝试给catalog[title]['available']赋值——此时catalog[title]尚未被创建为字典，因此会抛出KeyError（比如测试用例里的'Harry Potter'）。

def update_availability(catalog, title, new_availability):
    if title in catalog:
        catalog[title]['available'] = new_availability
    else:
        catalog[title]['available'] = False  # 此处触发KeyError，因catalog[title]不存在

修复方案 #

当书籍不存在时，需要先为该书名创建一个包含available键的字典条目，再完成赋值：

def update_availability(catalog, title, new_availability):
    if title in catalog:
        catalog[title]['available'] = new_availability
    else:
        # 初始化新书籍条目，默认可用性为False
        catalog[title] = {'available': False}

也可以用更简洁的写法，借助字典的get方法简化逻辑：

def update_availability(catalog, title, new_availability):
    # 书籍存在则取现有字典，不存在则创建空字典
    book_entry = catalog.get(title, {})
    # 根据书籍是否存在设置对应可用性
    book_entry['available'] = new_availability if title in catalog else False
    # 将更新后的条目放回目录
    catalog[title] = book_entry

验证示例 #

# 初始化图书馆目录
library = {"1984": {"available": True}, "Animal Farm": {"available": False}}

# 更新已有书籍的可用性
update_availability(library, "1984", False)
print(library["1984"]["available"])  # 输出: False

# 处理不存在的书籍
update_availability(library, "Harry Potter", True)
print(library["Harry Potter"]["available"])  # 输出: False

内容的提问来源于stack exchange，提问作者3rdKing-