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

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

Answer 2

一般来说，如果您需要在对函数的一次调用和下一次调用之间保留状态，您需要的是对象（您的解决方案）或生成器。在某些情况下，一种会比另一种更简单，但原则上您的操作方式并没有错误（尽管您似乎在实现方面存在一些问题）。

Sven 的建议，itertools.count()，是一个生成器。它的实现是这样的：

def count():
    i = 0
    while True:
        yield i
        i += 1

现在，如果您希望它像函数一样可调用，而不必执行 next(c)，您可以定义一个包装器来实现它：

 def count(c=itertools.count()):
     return next(c)

或者不可避免的单行 lambda：

count = lambda c=itertools.count(): next(c)

然后count() 每次调用它都会返回下一个整数。

当然，如果您希望能够创建任意数量的可调用函数，每个函数都有自己的计数器，您可以为此编写一个工厂：

def counter():
    return lambda c=itertools.count(): next(c)

那就是：

c = counter()
print c()   # 0
print c()   # 1
# etc

在我看来，这仍然比一个对象简单，但也不简单。如果您的状态或逻辑更复杂，对象的封装可能会胜出。