如何重写字典中的字典关键字？

def rewrite_dict_keywords(dictionary, old_keyword, new_keyword):
    keys = list(dictionary.keys())  # 获取字典中的所有键
    for key in keys:
        value = dictionary.pop(key)  # 删除键值对并存储在临时变量中
        if key == old_keyword:
            key = new_keyword  # 使用新的关键字替换旧的关键字
        dictionary[key] = value  # 将键值对添加回字典中

# 示例用法
my_dict = {'name': 'John', 'age': 25, 'address': {'city': 'New York', 'country': 'USA'}}
print("原始字典:", my_dict)

rewrite_dict_keywords(my_dict, 'address', 'location')
print("重写后的字典:", my_dict)

原始字典: {'name': 'John', 'age': 25, 'address': {'city': 'New York', 'country': 'USA'}}
重写后的字典: {'name': 'John', 'age': 25, 'location': {'city': 'New York', 'country': 'USA'}}