如何制作一个字典,为字典中缺少的键返回键,而不是引发KeyError?

问题描述

我要创建一个python字典,该字典向我返回字典中缺少的键的键值。

使用示例:

dic = smart_dict()
dic['a'] = 'one a'
print(dic['a'])
# >>> one a
print(dic['b'])
# >>> b

解决方案

dict有一个__missing__挂钩:

class smart_dict(dict):
    def __missing__(self, key):
        return key

可以简化为(因为self从未使用过):

class smart_dict(dict):
    @staticmethod
    def __missing__(key):
        return key

相关文章