Pyomo添加索引集至约束报错“Cannot apply a Set operator...”求助

错误根源 #

你遇到的Cannot apply a Set operator to an indexed Set component (td)错误，核心是约束规则里的索引逻辑完全错误：

1. 约束规则的参数顺序与model.td的元组结构不匹配（td的元素是(r,i)，规则参数却写成了i,r）
2. 错误使用生成器表达式(r for (r,i) in model.td)作为PD变量的索引，这不符合Pyomo变量的索引规则
3. 代码中未定义model.TG变量，直接引用会导致额外错误

修正步骤与完整代码 #

1. 简化扁平集合生成

原td_flat_rule可以用列表推导式简化，逻辑更清晰：

def td_flat_rule(model):
    return [(r, i) for r in model.r for i in model.td_raw[r]]

2. 补充缺失的TG变量定义

根据约束逻辑，TG应该是按t索引的变量或参数，这里补充定义为非负实数变量：

model.TG = pyo.Var(model.t, domain=pyo.NonNegativeReals)

3. 修正约束规则

调整参数顺序，直接用r索引PD变量：

def PD_constraint_rule(model, r, i):
    return model.PD[r] >= model.TG[i]

4. 完整修正代码

import pyomo.environ as pyo
from pyomo.opt import SolverFactory

td = {1:[1,2,3,5],2:[4]}

model = pyo.ConcreteModel()
model.r = pyo.RangeSet(1,2)
model.t = pyo.RangeSet(1,5)

model.td_raw = pyo.Set(model.r, initialize=td)

# 简化扁平集合生成
def td_flat_rule(model):
    return [(r, i) for r in model.r for i in model.td_raw[r]]
model.td = pyo.Set(rule=td_flat_rule, dimen=2)

model.PD = pyo.Var(model.r, domain=pyo.NonNegativeReals)
# 补充定义TG变量
model.TG = pyo.Var(model.t, domain=pyo.NonNegativeReals)

# 修正约束规则参数顺序与索引逻辑
def PD_constraint_rule(model, r, i):
    return model.PD[r] >= model.TG[i]

model.AxbConstraint = pyo.Constraint(model.td, rule=PD_constraint_rule)

opt = SolverFactory('glpk')
results = opt.solve(model)
results.write()

关键说明 #

• 当用二维集合model.td作为约束的索引时，Pyomo会自动将每个元组(r,i)的元素拆解为规则的参数，所以规则参数顺序必须和元组元素顺序一致
• PD变量是按r索引的，直接通过model.PD[r]引用即可，无需额外的集合遍历逻辑
• 如果TG是已知参数而非变量，可改用pyo.Param(model.t, initialize=...)定义

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