python 魔法方法 __getitem__
在字典和列表中我们经常使用[]来获取字典中的值或者列表中的元素,是因为字典和列表都内置了__getitem方法。
"__getitem__" in (dir(list)) >>>True "__getitem__" in (dir(dict)) >>>True
我们自定义一个类,当实例对象通过[]运算符取值时,会自动调用它的__getitem__方法
class DataBase: def __init__(self,id,address): #初始化方法 self.id=id self.address=address self.d={self.id:1,self.address:"192.168.1.1"} def __getitem__(self,key): return self.d.get(key,"default") data=DataBase(1,"192.168.1.1") print(data['hi']) >>>"default" #data.d中无 “hi”键, 返回 “default” print(data[data.id]) >>>1 print(data[data.address]) >>>"192.168.1.1"
序列可以迭代,即 for i in P
内置方法__getitem__(self, index)同样可以实现对象迭代(从索引0开始)。
class DataBase: def __init__(self,id,address): #初始化方法 self.id=id self.address=address self.d={self.id:1,self.address:"192.168.1.1"} def __getitem__(self,key): assert key < 10 print(key) #打印索引值 return self.d.get(key,"default") #返回字典value,默认为default data=DataBase(1,"192.168.1.1") for i in data: print(i) >>>0 default 1 1 # 字典中含有key1, 输出value1 2 default 3 default 4 default 5 default 6 default 7 default 8 default 9 default
实现__iter__和__next__同样的效果
i_data = iter(data) next(i_data) >>>0 default