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

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

所以我得到 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' 变量。我肯定有更好的方法来做到这一点...

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

请您参考如下方法:

显而易见的做法是将 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 

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