Pyomo中含双索引变量的分段线性函数建模求助

在设施选址分配优化建模中，需为节点×车辆双索引变量应用不同的分段线性函数，以下是对原方案错误的分析及可行实现：

原方案错误分析 #

Option 1 问题 #

1. 参数顺序颠倒：Piecewise组件的pw_pts应为自变量（arriv）的断点，f_rule为对应函数值（y），原代码将字典的keys和values搞反
2. 变量引用错误：第三个Piecewise误使用arriv_cd/y_cd替代arriv_ef/y_ef
3. 变量冗余：无需按车辆类型定义多组重复变量，可通过规则统一处理

Option 2 问题 #

仅实现了因变量y的线性组合约束，缺失自变量arriv的断点凸组合约束，且错误将组合系数定义为二进制变量（应为非负连续变量，且和为1）

Option 3 问题 #

Piecewise是Pyomo的独立组件，不能直接作为表达式添加到ConstraintList中

正确实现方案 #

使用统一双索引变量，通过规则为不同车辆类型绑定对应分段函数，正确配置Piecewise组件：

完整代码 #

# 重新组织分段函数：x=自变量断点，y=对应函数值
piece_wise = {
    'ab': {'x': [0, 1, 2], 'y': [1, 1, 2]},
    'cd': {'x': [0, 1, 2], 'y': [1, 1, 3]},
    'ef': {'x': [0, 1, 2], 'y': [1, 2, 4]}
}

# 1. 定义统一双索引变量
model.arriv = Var(model.Nodes, model.Vehicles, 
                  within=NonNegativeIntegers, bounds=(0, max_chargers))
model.y = Var(model.Nodes, model.Vehicles, 
              within=NonNegativeIntegers, bounds=(0, max_chargers))

# 2. 编写Piecewise规则：为每个(节点, 车辆)对指定分段函数参数
def pw_rule(model, i, veh):
    if 'ab' in veh:
        pw_pts = piece_wise['ab']['x']
        f_vals = piece_wise['ab']['y']
    elif 'cd' in veh:
        pw_pts = piece_wise['cd']['x']
        f_vals = piece_wise['cd']['y']
    elif 'ef' in veh:
        pw_pts = piece_wise['ef']['x']
        f_vals = piece_wise['ef']['y']
    else:
        return Constraint.Skip  # 跳过未知车辆类型
    
    # 返回格式：(自变量, 因变量, 断点, 函数值, 表示方法, 约束类型)
    return (model.arriv[i, veh], model.y[i, veh], pw_pts, f_vals, 'SOS2', 'EQ')

# 3. 创建Piecewise组件
model.piecewise_constraints = Piecewise(model.Nodes, model.Vehicles, rule=pw_rule)

# 4. 修正服务约束：使用统一y变量
def service_rule(model, i, veh):
    try:
        return sum(model.arrival[o,d,veh] * model.lam[i,o,d,veh] for o,d in model.sp) <= model.y[i, veh]
    except KeyError:
        return Constraint.Skip

model.service = Constraint(model.Nodes, model.Vehicles, rule=service_rule)

关键说明 #

• 参数正确性：确保pw_pts是自变量的断点序列，f_vals是对应函数值序列，顺序严格匹配
• 规则复用：通过pw_rule自动为不同车辆类型绑定对应的分段函数，避免重复代码
• 组件用法：Piecewise组件会自动为每个索引对生成对应的分段线性约束，无需手动添加到ConstraintList

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