如何在Pyomo中对多索引对象执行单索引维度操作？

问题分析 #

你的代码存在两个核心问题：

1. 约束的索引集合与逻辑不匹配，触发索引错误（KeyError）
2. 求和循环范围错误，无法实现按年度维度的统计

面积约束错误点 #

• 约束需按每个年份单独构建，Constraint的索引集合应为model.t（年份集合），而非代码里的model.a/model.n（集合命名存在不一致，变量用model.crop/model.t，约束却用了其他集合）
• 求和循环重复遍历年份集合，导致model.x[crop, year]的索引冲突，引发KeyError

用水约束错误点 #

• 用水约束是年度总用水量限制，无需按作物索引构建，仅需按年份索引即可
• 求和循环错误遍历所有作物和年份，导致累加了全周期用水量，而非单一年度

修正后的代码 #

1. 面积约束（每年总种植面积限制） #

def area_constr(model, year):
    # 固定当前年份，遍历所有作物求和
    return sum(model.x[crop, year] for crop in model.crop) <= model.total_cropping_area_n[year]

# 按年份集合构建约束，每个年份对应一条约束
model.area_constr = pyomo.Constraint(model.t, rule=area_constr, doc='Restricción de superficie disponible')

2. 用水约束（每年总用水量限制） #

def water_constr(model, year):
    # 固定当前年份，遍历所有作物计算年度总用水量
    return sum(model.x[crop, year] * model.crop_water_demand[crop] for crop in model.crop) <= model.total_water_endowment[year]

# 按年份集合构建约束
model.water_constr = pyomo.Constraint(model.t, rule=water_constr)

关键说明 #

• 操作多索引对象的单维度时，只需固定目标维度的索引值，遍历其他维度即可完成对应维度的聚合计算
• 约束的索引集合必须与逻辑匹配：如果是年度维度限制，约束就按年份集合索引，而非多维度索引
• 注意集合命名一致性：代码中存在model.crop/model.t与model.a/model.n的命名冲突，需确保所有集合引用统一

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