Question

如何编写一个只在每次调用时返回递增数字的函数？

print counter() # 0
print counter() # 1
print counter() # 2
print counter() # 3
etc

为清楚起见，我不是在寻找发电机（尽管答案可能会使用它）。该函数应返回一个整数，而不是一个可迭代的对象。我也不是在寻找涉及全局或其他共享变量（类，函数属性等）的解决方案。

请点击此处http://www.valuedlessons.com/2008/01/monads-in-python-with-nice-syntax.html了解有关此事的一些有用见解。

Answer 1

您不需要实现该功能 - 它已经存在：

counter = itertools.count().next

或在Python 3.x中

counter = itertools.count().__next__

更一般地说，如果你想要一个具有附加状态的可调用，有几种解决方案：

一堂课：

class Counter(object):
    def __init__(self, start=0):
        self.count = 0
    def __call__(self):
        val = self.count
        self.count += 1
        return val

一个可变的默认参数：

def counter(_count=[0]):
    val = _count[0]
    _count[0] += 1
    return val

关闭：

def make_counter(start=0):
    count = [start]
    def counter():
        val = count[0]
        count[0] += 1
        return val
    return counter

在Python 3.x中，你不再需要使用上面的hack，因为你可以使用nonlocal声明：

def make_counter(start=0):
    count = start
    def counter():
        nonlocal count
        val = count
        count += 1
        return val
    return counter

函数参数：

def counter():
    val = counter.count
    counter.count += 1
    return val
counter.count = 0

发电机：

def count(start=0):
    while True:
        yield start
        start += 1
counter = count().next   ## or .__next__ in Python 3.x

当然，所有这些解决方案都必须将当前计数存储在某处。

Answer 2

所以，你想编写一个通过不使用任何语言功能来完成魔术的函数，对吧？那是胡说八道。你要么编写一个生成器，要么创建一个可调用的类（实例属性不是由任何东西“共享”）。

class Counter(object):
    def __init__(self):
        self.n = -1
    def __call__(self):
        self.n += 1
        return self.n

counter = Counter()
counter() # 0
# ...

# generator
def counter():
    n = 0
    while True:
        yield n
        n += 1

Answer 3

简而言之，

from itertools import count
counter = lambda c=count(): next(c)

简单数字生成器

3 个答案: