Python dict重写get方法

Python dict重写get方法

一.应用场景

有时,在使用get方法获取对应的value的时候,如:df_dict.get(“name”),
如果name对应的value为None或者name这个key不存在时.希望将这个key加入到df_dict中,并且给value一个值;
此时可以对df_dict这个对象的get方法进行重写.

二.步骤

1.自定义一个MyDict的类,并继承自dict类;
2.重写get方法;
3.实例化MyDict创建df_dict对象;
4.df_dict的其它用法与dict一样;
5.当使用get方法时,进入重写方法;

三.示例代码

class MyDict(dict):
    """重写dict"""

    def __init__(self,kwargs):
        super().__init__(kwargs)

    def get(self,k, d=None):
        # 自己处理的逻辑,将自定义处理后的结果替换赋值给d即可
        return self[k] if k in self else d