网络编程
位置:首页>> 网络编程>> Python编程>> 详解Python中的__getitem__方法与slice对象的切片操作

详解Python中的__getitem__方法与slice对象的切片操作

作者:cangmean  发布时间:2022-04-26 19:54:55 

标签: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对象的话,就创建一个实例并返回。

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com