__getitem__

2023-01-31 00:01:12

实现了__iter__()的实例虽能用于for循环,看似像list,但并不能将其当做list来使用,比如,Fib()[5]还是报错
可通过实现__getitem__()方法,;来实现让实例像list那样按照下标取出元素

  实现按照下标取元素

    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

 

  实现切片功能

  __getitem__()传入的参数可能是一个int,也可能是一个切片对象slice,因此需要做判断
  如果把对象看成dict,__getitem__()的参数也可能是一个可以作key的object,例如str

    class Fib(object):
        def __getitem__(self, n):
            if isinstance(n, int): # n是索引
                a, b = 1, 1
                for x in range(n):
                    a, b = b, a + b
                return a
            if isinstance(n, slice): # n是切片
                start = n.start
                stop = n.stop
                if start is None:
                    start = 0
                a, b = 1, 1
                L = []
                for x in range(stop):
                    if x >= start:
                        L.append(a)
                    a, b = b, a + b
                return L
            
    f[0:5]    #输出:[1, 1, 2, 3, 5]
    f[:10]    #输出:[1, 1, 2, 3, 5, 8, 13, 21, 34, 55]
    #该例没有对step参数作处理,也没有对负数作处理,所以,要正确实现一个__getitem__()还是有很多工作要做的

 

  __setitem__和__delitem__

  与之对应的还有__setitem__()方法和__delitem__()方法,分别用于为某个元素设值和删除某个元素
  总之,通过上面的方法,可自定义的类表现得和python自带的list、tuple、dict没什么区别,这完全归功于动态语言的"鸭子类型",不需要强制继承某个接口

 

相关文章