Python中是否支持通过.next(arg)传参的参数化生成器？

当然可行！不过得先纠正一个小细节：Python的生成器并没有.next(arg)这种调用方式，你需要使用send()方法来向生成器传递参数——next(gen)本质上就等价于gen.send(None)。下面我给你提供一个完全符合需求的实现：

核心实现代码 #

def param_generator(initial_value=0):
    current = initial_value
    while True:
        # 接收外部传入的参数，首次启动时这里会返回None
        arg = yield current
        if arg is True:
            current += 1
        elif arg is False:
            current -= 1
        # 可选：处理非布尔参数的情况
        # else:
        #     raise ValueError("Generator only accepts True/False as arguments")

使用示例 #

# 初始化生成器，初始值设为0（也可以自定义初始值，比如param_generator(5)）
gen = param_generator()

# 首次调用必须用next()或send(None)来启动生成器，此时返回初始值
print(next(gen))  # 输出: 0

# 传递True，让数值加1
print(gen.send(True))  # 输出: 1

# 传递False，让数值减1
print(gen.send(False))  # 输出: 0

# 再次传递True
print(gen.send(True))  # 输出: 1

原理说明 #

• 生成器在执行到yield current时会暂停，并返回当前的current值。
• 当你调用send(arg)时，这个arg会被赋值给yield表达式的返回值（也就是代码里的arg变量），然后生成器继续执行逻辑（加1或减1），直到下一次yield暂停，返回更新后的current。
• 首次调用必须用next()或者send(None)，因为此时生成器还没走到第一个yield的位置，无法接收参数。

如果需要更严谨的参数校验，可以取消代码里的注释，添加对非布尔参数的异常抛出逻辑，避免传入无效参数导致意外行为。

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