python 将对象设置为可迭代的两种实现方法
时间:2022-05-13 13:50:22|栏目:Python代码|点击: 次
1、实现 __getitem__(self)
class Library(object):
def __init__(self):
self.value=['a','b','c','d','e']
def __getitem__(self, i):
if i>=len(self.value):
raise IndexError("out of index")
value=self.value[i]
return value
调用的时候,系统默认从0 开始传入,并使得i=i+1
2、实现 __iter__(self),next(self)
class Library2(object):
def __init__(self):
self.value=['a','b','c','d','e']
self.i=-1
def __iter__(self):
return self
def next(self):
self.i += 1
if self.i>=len(self.value):
raise StopIteration
return self.value[self.i]
test=Library2()
print test.next()
print test.next()
在这里可以像生成器一样使用
栏 目:Python代码
本文地址:http://www.codeinn.net/misctech/201734.html






