详解Python中的__getitem__方法与slice对象的切片操作
更新时间:2016年06月27日 15:12:19 作者:cangmean
Python中想要使类的实例像list一样使用下标,可以用__getitem__方法,而配合slice对象则可以实现list一样的切片,详解Python中的__getitem__方法与slice对象的切片操作
Fib实例虽然能作用于for循环,看起来和list有点像,但是,把它当成list来使用还是不行,比如,取第5个元素:
>>> Fib()[5] Traceback (most recent call last): File "<stdin>", line 1, in <module> TypeError: 'Fib' object does not support indexing
要表现得像list那样按照下标取出元素,需要实现__getitem__()方法:
class Fib(object): def __getitem__(self, n): a, b = 1, 1 for x in range(n): a, b = b, a + b return a
现在,就可以按下标访问数列的任意一项了:
>>> f = Fib() >>> f[0] 1 >>> f[1] 1 >>> f[2] 2 >>> f[3] 3 >>> f[10] 89 >>> f[100] 573147844013817084101
slice对象与__getitem__
想要使类的实例像列表一样使用下标, 可以设置__getitem__方法。比如:
class _List(object): def __getitem__(self, key): print key l = _List() l[3] # print 3
但是如果想要使用切片操作的
l[1:4] # print slice(1, 4, None)
会创建一个slice对象用于切片, 可以通过help(slice)查看具体操作。
a = slice(1, 4, None) range(5)[a] # print [1, 2, 3]
更加丰富的操作
class _List(object): def __init__(self, _list): self._list = _list def __getitem__(self, key): if isinstance(key, int): return self._list[key] elif isinstance(key, slice): reutrn self.__class__(self._list[key]) if __name__ == '__main__': c = _List(range(10)) b = c[1:5] print b[3] # print 4
如果key是一个整形的话就返回列表元素,如果是一个slice对象的话,就创建一个实例并返回。
相关文章
对python内置map和six.moves.map的区别详解
今天小编就为大家分享一篇对python内置map和six.moves.map的区别详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
最新评论