实现了__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没什么区别,这完全归功于动态语言的"鸭子类型",不需要强制继承某个接口