python - 类中的方法可以是生成器吗？

Question

将类中的方法用作生成器是否可接受/Pythonic？我找到的所有示例都在函数中显示了 yield 语句，而不是在类中。

这是一个示例工作代码:

class SomeClass(object):
    def first_ten(self):
        for i in range(10):
            yield i

    def test(self):
        for i in self.first_ten():
            print i

SomeClass().test()

Answer 1

是的，这是完全正常的。例如，它通常用于实现 object.__iter__() method。使对象成为可迭代对象:

class SomeContainer(object):
    def __iter__(self):
        for elem in self._datastructure:
            if elem.visible:
                yield elem.value

但是，不要受限于该通用模式；任何需要迭代的东西都是生成器方法的候选对象。