详解Python中的__getitem__方法与slice对象的切片操作
Fib实例虽然能作用于for循环,看起来和list有点像,但是,把它当成list来使用还是不行,比如,取第5个元素:
>>>Fib()[5] Traceback(mostrecentcalllast): File"<stdin>",line1,in<module> TypeError:'Fib'objectdoesnotsupportindexing
要表现得像list那样按照下标取出元素,需要实现__getitem__()方法:
classFib(object): def__getitem__(self,n): a,b=1,1 forxinrange(n): a,b=b,a+b returna
现在,就可以按下标访问数列的任意一项了:
>>>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): printkey l=_List() l[3]#print3
但是如果想要使用切片操作的
l[1:4]#printslice(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): ifisinstance(key,int): returnself._list[key] elifisinstance(key,slice): reutrnself.__class__(self._list[key]) if__name__=='__main__': c=_List(range(10)) b=c[1:5] printb[3]#print4
如果key是一个整形的话就返回列表元素,如果是一个slice对象的话,就创建一个实例并返回。