python - 一个自动递增的可调用对象？

Question

我有一个方法，我需要在其中将一个不断增加的整数传递给另一个函数。

我可以这样做:

def foo(i):
    print i

def bar():

    class Incrementer(object):
        def __init__(self, start=0):
            self.i = start

        def __get__(self):
            j = self.i
            self.i += 1
            return j

    number = Incrementer()
    foo(number)
    foo(number)
    foo(number)

正确输出 0 1 2 ... 但我觉得我忽略了一种更简单(或内置)的方法？

Answer 1

尝试 itertools.count() -- 它完全满足您的需求:

>>> c = itertools.count()
>>> next(c)
0
>>> next(c)
1
>>> next(c)
2