Pyomo在约束中自动解包namedtuple的原因及修复方案

你猜的没错——Pyomo会自动解包可迭代类型的集合元素（比如tuple、namedtuple），把每个字段拆成独立参数传给约束规则函数。你的Product是namedtuple，有2个字段，所以Pyomo会给rule_product_quantity传3个参数：model加上解包后的2个字段，但你的函数只定义了2个参数，自然就报参数不匹配的错了。

这个设计是为了方便处理多维索引的集合。比如当你用tuple表示多维索引（像("product1", "factoryA")这种）时，规则函数可以直接拿到每个维度的参数，不用自己手动去拆元组。举个例子：

# 多维集合示例
prod_factory = pyo.Set(initialize=[("product1", "factoryA"), ("product2", "factoryB")])
def supply_rule(model, prod, factory):
    # 直接用prod和factory，不用写idx[0]、idx[1]
    return model.supply[prod, factory] <= model.capacity[factory]

这样写起来比手动解包要简洁很多。

最省事还能保持代码封装性的方法，是给你的Set显式指定dimen=1，告诉Pyomo把每个元素当成一个整体，不要解包：

from collections import namedtuple
import pyomo.environ as pyo

Product = namedtuple("Product", "product quantity")
# 添加dimen=1，指定集合维度为1，元素不被解包
products = pyo.Set(initialize=[Product("product1", 10), Product("product2", 20)], dimen=1)
model = pyo.ConcreteModel()
model.product_quantity = pyo.Var(products, within=pyo.NonNegativeReals)

def rule_product_quantity(model, prod):
    return model.product_quantity[prod] >= prod.quantity

model.ct_product_lim = pyo.Constraint(products, rule=rule_product_quantity)

要是你不想改集合的定义，也可以调整规则函数的参数，接受解包后的字段：

def rule_product_quantity(model, product_name, quantity):
    # 重新组装成Product对象来索引变量
    prod = Product(product_name, quantity)
    return model.product_quantity[prod] >= quantity

不过这种方法需要手动拼回Product对象，不如第一种方法简洁。

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