可以使用字典推导式来编写该函数,实现如下:
def extract_keys(dictionary, selected_keys):
return {k: v for k, v in dictionary.items() if k in selected_keys}
my_dict = {'name': 'Tom', 'age': 25, 'address': 'Beijing'}
selected_keys = ['name', 'address']
result = extract_keys(my_dict, selected_keys)
print(result) # 输出结果为:{'name': 'Tom', 'address': 'Beijing'}
函数接受两个参数:一个字典和一个选定的键列表。函数返回一个新的字典,仅包含原字典中选定的键。我们使用字典推导式创建了新字典,其中仅包含选定的键。最后,我们调用函数并将其结果打印出来,以验证其功能。