如何使用Python的魔法方法__getitem__、__setitem__和__delitem__实现自定义数据结构？

本文共计520个文字，预计阅读时间需要3分钟。

1. 使用 `__getitem__` 魔法函数对于字典 `a={name: zhangsan, age: 20}`，可以通过 `a[name]` 取值。之所以可以这样用，是因为 dict 类实现了 `__getitem__` 魔法函数。同样，在我们自定义的类中，也可以通过实现 `__getitem__` 来实现类似的功能。

1、__getitem__魔法函数

• 对于字典比如a={“name”:“zhangsan”,“age”:20}，可以通过a[“name”]取值，之所以可以这么用，就是因为dict的类实现了__getitem__魔法函数，同样，在我们自定义的类中，如果我们也实现了这个魔法函数，那么自定义类的对象同样也可以像字典那样取值
  如：

class Student():
def __init__(self):
self.attr={"name":"zhangsan","age":20}

def __getitem__(self, item):
return self.attr[item]

s=Student()
print(s["name"])

执行结果为：

zhangsan

2、__setitem__魔法函数

• 同样如果一个类如果实现了__setitem__魔法函数，则这个类的对象就可以像给一个字典设置key-value一样的方式设置值，如：

class Student():
def __init__(self):
self.attr={"name":"zhangsan","age":20}

def __getitem__(self, item):
return self.attr[item]

def __setitem__(self, key, value):
self.attr[key]=value

s=Student()
s["name"]="张三丰"
s["age"]=120
print(s["name"])
print(s["age"])

执行结果如下：

张三丰
120

3、__delitem__魔法函数

• 如果一个类实现了__delitem__，则这个类的对象就可以使用类似 del obj[“name”]的操作
  如：

class Student():
def __init__(self):
self.attr={"name":"zhangsan","age":20}

def __getitem__(self, item):
return self.attr[item]

def __setitem__(self, key, value):
self.attr[key]=value

def __delitem__(self,key):
del self.attr[key]

s=Student()
s["name"]="张三丰"
s["age"]=120
print(s["name"])
print(s["age"])
del s["name"]
print(s["name"])

此时执行结果如下：

张三丰
120
Traceback (most recent call last):
File "G:\src\django\test.py", line 20, in <module>
print(s["name"])
File "G:\src\django\test.py", line 6, in __getitem__
return self.attr[item]
KeyError: 'name'