python - 在输入不变的情况下在python中生成连续数字

Question

我需要在输入数字不变的情况下获取连续数字。

所以我得到 give(5)->1，give(5)->2，依此类推，然后:再次 give(6)->1，开始计数。

到目前为止，我用迭代器函数 count() 和函数 give(num) 解决了这个问题，如下所示:

def count(start=1):
    n=start
    while True:
        yield n
        n +=1
def give(num):
    global last
    global a
    if num==last:
        ret=a.next()
    else:
        a=count()
        ret=a.next()
    last=num
    return ret

它有效，但它很丑陋:我有两个全局变量，必须在调用 give(num) 之前设置它们。我希望能够调用 give(num) 而无需事先设置 'a=count()' 和 'last=999' 变量。我肯定有更好的方法来做到这一点...

编辑:感谢大家对令人难以置信的快速和多样的回应，我在这里有很多东西要学习......

Answer 1

显而易见的做法是将 give 变成一个对象而不是一个函数。* 任何对象都可以通过定义一个 __call__ 来调用。方法。

当我们这样做时，您的代码可以大大简化，所以让我们这样做吧。

class Giver(object):
    def __init__(self):
        self.last, self.a = object(), count()
    def __call__(self, num):
        if num != self.last:
            self.a = count(1)
        self.last = num
        return self.a.next()

give = Giver()

所以:

>>> give(5)
1
>>> give(5)
2
>>> give(6)
1
>>> give(5)
1

这还允许您创建多个独立的给予者，每个给予者都有自己独立的当前状态，如果您有任何需要的话。

如果你想用更多的状态扩展它，状态就进入实例变量。例如，您可以将 last 和 a 替换为将先前看到的值映射到计数器的字典:

class Giver(object):
    def __init__(self):
        self.counters = defaultdict(count)
    def __call__(self, num):
        return next(self.counters[num])

现在:

>>> give(5)
1
>>> give(5)
2
>>> give(6)
1
>>> give(5)
3

---

* 我在这里跳过了一步。您始终可以通过将变量和使用它们的所有内容(可能只是一个函数)放在函数或其他范围内来删除全局变量，因此它们最终作为函数闭包中的自由变量。但在你的情况下，我认为这只会让你的代码看起来“更丑陋”(在同样的意义上你认为它很丑陋)。但请记住，对象和闭包在它们能做的事情上实际上是等同的，但在它们的外观上是不同的——所以当一个看起来非常丑陋时，试试另一个。