python 将对象设置为可迭代有两种实现方式

来源:互联网 发布:ug汽车大模具编程刀路 编辑:程序博客网 时间:2024/06/06 07: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()
  
  在这里可以像生成器一样使用
0 0
原创粉丝点击