python – 定义一个支持两个或多个同时迭代的可迭代类

2024年2月5日 238次阅读

我不确定这个标题是不是很好.希望有人可以帮忙改变它.

我正在尝试定义一个分子类,并希望它可以迭代它的原子.
我搜索了人们如何定义一个可迭代类,它可能看起来像这样：

class molecule(object):
    def __init__(self,name):
         self.__pointer=0
         # ...something else
    def __iter__(self):
         self.__pointer=0
         return self
    def __next__(self):
         self.__pointer+=1
         if self.__pointer<=self.natoms: # self.natoms is defined somewhere
               return self[self.__pointer]   # suppose __getitem__ is defined to return an atom object
         else:
               raise StopIteration("Exceeded all atoms")

效果很好：

 >>> ben=molecule('ben') #defined a molecule
 >>> ben.addatom(...) # and defined some atoms
 >>> ite=iter(ben)
 >>> for atom in ite:
 ...     print(atom)
 ...
 # atom objects are correctly printed here

但是,我发现它不能在两个迭代器中工作,如果它们同时存在的话.

>>> ite=iter(ben)
>>> next(ite)
# atom1
>>> next(ite)
# atom2
>>> ite2=iter(ben)
>>> next(ite)
# atom1 again, where atom3 is expected
>>> next(ite2)
# atom2, where atom1 is expected

这并不奇怪,因为两个迭代器共享一个相同的self .__指针,因此定义一个新的迭代器会将指针更新为零.

我看了一下这个页面How to make class iterable?,大多数都在课堂上使用了自我.__指针,这引起了我的质疑.
我猜如果指针是迭代器(ite或ite2)的属性而不是迭代对象本身(分子)的属性,那么这个问题就可以解决了.

希望有人能给予一些帮助:)谢谢.

最佳答案 每次调用时都使__iter__返回新的迭代器对象.

class molecule_iterator(object):
    def __init__(self, obj):
        self.pointer = 0
        self.obj = obj

    def __next__(self):
        self.pointer += 1  # Are you sure to do this here?
        if self.pointer < self.obj.natoms:
            return self.obj[self.pointer]
        raise StopIteration()


class molecule(object):
    ...

    def __iter__(self):
        return molecule_iterator(self)

    ...

或者,您可以使用yield statement,它不需要您定义__next__.因为__iter __()将返回一个新的生成器迭代器：

def __iter__(self):
    for i in range(self.natoms):
        yield self[i + 1]

DEMO：https://ideone.com/TK78Ml

点赞