详解Python中的__getitem__方法与slice对象的切片操作-Python教學-PHP中文網

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 slice 切片

來源：php.cn

上一篇：理解生产者消费者模型及在Python编程中的运用实例下一篇：浅谈python中scipy.misc.logsumexp函数的运用场景

本網站聲明

本文內容由網友自願投稿，版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容，請聯絡admin@php.cn

作者最新文章