如何在Python中实现纯声明式可参数化的无限数字序列迭代器

最优方案：使用标准库内置itertools.count #

Python标准库itertools模块已经原生提供了完全符合要求的无限迭代器实现，仅需1行导入即可使用：

from itertools import count

特性完全匹配需求： #

• 纯函数式：无外部副作用，迭代状态完全封装在迭代器实例内部，不修改任何全局变量
• 可参数化：支持自定义start（起始值）、step（步长）参数，适配任意计数场景
• 声明式：直接定义迭代规则，无冗余的协议实现代码
• 极高可读性：Python开发者无需额外心智负担即可理解和使用

适配原有测试fixture的简化版代码： #

import pytest
from itertools import count

@pytest.fixture
def create_user_model_test_data():
    num_iterator = count(start=0)
    for n in num_iterator:
        if n == 20:
            return
        create_user(name=f"user-{n}", role=["admin", "user"][n % 2])

手动实现版（无标准库依赖） #

如果需要自行实现逻辑，使用Python生成器即可，代码同样简洁易读：

def infinite_counter(start: int = 0, step: int = 1):
    current = start
    while True:
        yield current
        current += step

使用方式和itertools.count完全一致，同样满足所有要求。

补充说明 #

Python原生对函数式编程场景的支持足够覆盖绝大多数业务需求，标准库内置了大量成熟的函数式工具，兼顾了简洁性和工程可维护性，完全无需为了基础的迭代器需求切换技术栈。创业公司团队规模小，使用成熟度高、生态完善的技术栈能够大幅降低维护成本，提升迭代效率。

内容的提问来源于stack exchange，提问作者Pablo