如何使自定义对象可迭代？

Question

我有一个list自定义类对象（下面是示例）。

使用：list(itertools.chain.from_iterable(myBigList))我想将所有stations子列表“合并”到一个大列表中。因此，我认为我需要使自定义类成为可迭代的。

这是我的自定义类的示例。

class direction(object) :
    def __init__(self, id) :
        self.id = id              
        self.__stations = list()

    def __iter__(self):
        self.__i = 0                #  iterable current item 
        return iter(self.__stations)

    def __next__(self):
        if self.__i<len(self.__stations)-1:
            self.__i += 1         
            return self.__stations[self.__i]
        else:
            raise StopIteration

我实现了__iter__，__next__但是似乎没有用。他们甚至没有被召集。

知道我做错了什么吗？

注意：使用Python 3.3

Answer 1

__iter__ 当您尝试遍历类实例时调用的是什么：

>>> class Foo(object):
...     def __iter__(self):
...         return (x for x in range(4))
... 
>>> list(Foo())
[0, 1, 2, 3]

__next__是在返回的对象上调用的内容__iter__（在python2.x上next，不是__next__，我一般都对它们都使用别名，以便代码可以与其中任何一个一起使用…）：

class Bar(object):
   def __init__(self):
       self.idx = 0
       self.data = range(4)
   def __iter__(self):
       return self
   def __next__(self):
       self.idx += 1
       try:
           return self.data[self.idx-1]
       except IndexError:
           self.idx = 0
           raise StopIteration  # Done iterating.
   next = __next__  # python2.x compatibility.