详解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对象的话，就创建一个实例并返回。