Pyomo中RangeSet默认值导致约束判断异常的解决方法咨询

你的核心问题是约束判断逻辑错误：你用车道总数m.i.value和车辆数number_of_vehicle做比较，而非判断当前迭代的车道索引i是否为集合model.I的最后一个元素，导致所有约束都被跳过。此外，当i是最后一个车道时，i+1不在model.I的范围内，直接访问会引发索引错误，因此需要针对最后一个i跳过约束。

修正步骤 #

1. 调整判断条件：将错误的比较逻辑改为判断当前i是否为集合model.I的最后一个元素，即i < m.i.value，这样仅对非最后一个车道应用约束，最后一个车道直接跳过。
2. 确保索引合法性：只有当i不是最后一个元素时，i+1才会在model.I的范围内，避免索引越界。

修正后的完整代码 #

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

model = pyo.AbstractModel()

number_of_lane = 2
number_of_vehicle = 2

model.i = pyo.Param(within=pyo.NonNegativeIntegers, default=number_of_lane)
model.j = pyo.Param(within=pyo.NonNegativeIntegers, default=number_of_vehicle)

model.I = pyo.RangeSet(1, model.i)
model.J = pyo.RangeSet(1, model.j)

model.R = pyo.Param(default=0.5) # CAV反应时间(s)
model.D = pyo.Param(default=1.5) # 安全距离(m)
model.lv = pyo.Param(default=4) # 车辆长度(m)

# 补充xr_cons的示例初始化（原代码未定义，此处给出可运行的示例）
def xr_cons(m, i, j):
    return 0.0

model.xr = pyo.Param(model.I, model.J, within=pyo.NonNegativeIntegers, initialize=xr_cons)
model.x = pyo.Var(model.I, model.J, domain=pyo.NonNegativeReals, initialize=0)

def lane_crossing_constraint_rule(m, i, j):
    # 判断当前i是否为最后一个车道，是则跳过约束
    if i < m.i.value:
        return (m.x[i,j] - m.xr[i,j])**2 + (m.x[i+1,j] - m.xr[i+1,j])**2 >= (m.lv + m.D)
    else:
        return pyo.Constraint.Skip

model.l1Constraint = pyo.Constraint(model.I, model.J, rule=lane_crossing_constraint_rule)

关键修改说明 #

• 判断逻辑修正：用i < m.i.value替代原错误条件，确保仅对除最后一个车道外的所有i生成约束，避免i+1索引越界。
• 补充xr_cons定义：原代码中xr_cons未定义，此处添加示例初始化函数，保证代码可正常运行。

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