详解Python魔术方法__getitem__、__setitem__、__delitem__、__len__

Python的魔术方法一般以__methodname__的形式命名,如:__init__(构造方法), __getitem__、 __setitem__(subscriptable所需method), __delitem__(del obj[key]所需method), __len__(len(…)所需method)等。

在Python中,如果我们想实现创建类似于序列和映射的类,可以通过重写魔法方法__getitem__、__setitem__、__delitem__、__len__方法去模拟。


魔术方法的作用:

__getitem__(self,key):返回键对应的值。

__setitem__(self,key,value):设置给定键的值

__delitem__(self,key):删除给定键对应的元素。

__len__():返回元素的数量


代码实例:

# coding:utf-8
'''
    desc:尝试定义一种新的数据类型
          等差数列
    author:pythontab.com
'''
class ArithemeticSequence(object):
    def __init__(self,start=0,step=1):
        print 'Call function __init__'
        self.start=start
        self.step=step
        self.myData={}
    # 定义获取值的方法
    def __getitem__(self,key):
        print 'Call function __getitem__'
        try:
            return self.myData[key]
        except KeyError:
           return self.start+key*self.step
    # 定义赋值方法
    def __setitem__(self,key,value):
        print 'Call function __setitem__'
        self.myData[key]=value
    # 定义获取长度的方法
    def __len__(self):
        print 'Call function __len__'
        # 这里为了可以看出__len__的作用, 我们故意把length增加1
        return len(self.myData) + 1
    # 定义删除元素的方法
    def __delitem__(self, key):
        print 'Call function __delitem__'
        del self.myData[key]
    
s=ArithemeticSequence(1,2)
print s[3]  # 这里应该执行self.start+key*self.step,因为没有3这个key
s[3] = 100  # 进行赋值
print s[3]  # 前面进行了赋值,那么直接输出赋的值100
print len(s) # 我们故意多加了1,应该返回2
del s[3] # 删除3这个key
print s[3] # 这里应该执行self.start+key*self.step,因为3这个key被删了

输出结果:

Call function __init__
Call function __getitem__
7
Call function __setitem__
Call function __getitem__
100
Call function __len__
2
Call function __delitem__
Call function __getitem__
7


这些魔术方法的原理就是:当我们对类的属性item进行下标的操作时,首先会被__getitem__()、__setitem__()、__delitem__()拦截,从而进行我们在方法中设定的操作,如赋值,修改内容,删除内容等等。

上一篇:python类中super()和__init__()的区别
下一篇:Python中enumerate函数用法详解

PythonTab微信公众号:

Python技术交流互助群 ( 请勿加多个群 ):

群1: 87464755

群2: 333646237

群3: 318130924

群4: 385100854