在 Pyomo 中，由于表达式的不可变性，直接修改约束表达式并非易事。然而，通过结合 Expression 组件，我们可以实现类似于 Pulp 中逐步构建约束的效果。本文将深入探讨这种方法，并提供示例代码和注意事项，帮助读者更好地理解和应用。
使用 Expression 组件动态扩展约束
Pyomo 的核心设计理念是表达式的不可变性。这意味着一旦创建了一个表达式，其结构就不能被更改。这给动态修改约束带来了挑战，因为添加变量或修改系数实际上需要创建一个新的表达式树。
Expression 组件提供了一种巧妙的解决方案。它可以被视为一个“指针”，当它在表达式中使用时，我们可以修改这个指针指向的内容，而无需重建整个表达式树。
以下是一个示例，展示了如何使用 Expression 组件来逐步构建约束：
from pyomo.environ import ConcreteModel, Var, Constraint, Expression model = ConcreteModel() model.x = Var() model.Cons1_body = Expression(expr=0) # 初始化表达式为 0 model.Cons1 = Constraint(expr=model.Cons1_body == 200) model.Cons1_body += model.x * 2 # 向表达式中添加变量和系数 model.pprint()
这段代码首先创建了一个空的 Expression 对象 model.Cons1_body，并将其初始化为 0。然后，我们使用这个 Expression 对象来定义约束 model.Cons1。最后，我们通过 += 运算符向 model.Cons1_body 中添加变量 x 和系数 2，从而动态地扩展了约束。
model.pprint() 的输出结果如下：
1 Var Declarations
x : Size=1, Index=None
Key : Lower : Value : Upper : Fixed : Stale : Domain
None : None : None : None : False : True : Reals
1 Expression Declarations
Cons1_body : Size=1, Index=None
Key : Expression
None : 2*x
1 Constraint Declarations
Cons1 : Size=1, Index=None, Active=True
Key : Lower : Body : Upper : Active
None : 200.0 : Cons1_body : 200.0 : True
3 Declarations: x Cons1_body Cons1可以看到，约束 Cons1 已经成功地扩展为 2*x == 200。
替换约束表达式
虽然 Expression 组件允许我们动态地修改约束的内部表达式，但有时我们可能需要完全替换约束的表达式。Pyomo 提供了相应的机制来实现这一点：
from pyomo.environ import ConcreteModel, Var, Constraint model = ConcreteModel() model.x = Var() model.y = Var() model.Cons1 = Constraint(expr = model.x*2 == 200) # 注意：这里需要重新赋值 Constraint，而不是修改 Constraint.expr model.Cons1 = Constraint(expr = model.x*2 + model.y * 3 == 200) #修改了右端项 model.pprint()
需要注意的是，我们不能直接修改 model.Cons1.expr，而是需要重新赋值 Constraint 对象。
关于 "空" 约束的注意事项
Pyomo 不允许定义没有任何变量的约束。尝试创建 Constraint(expr=0 == 200) 会导致错误。
ERROR: Rule failed when generating expression for Constraint Cons1 with index None: ValueError: Invalid constraint expression. The constraint expression resolved to a trivial Boolean (False) instead of a Pyomo object. Please modify your rule to return Constraint.Infeasible instead of False.
但是，我们可以使用元组表示法来创建一个类似于 "空" 约束的对象：
from pyomo.environ import ConcreteModel, Constraint model = ConcreteModel() model.Cons2 = Constraint(expr=(200, 0, 200)) model.Cons2.pprint()
输出结果如下：
Cons2 : Size=1, Index=None, Active=True
Key : Lower : Body : Upper : Active
None : 200.0 : 0.0 : 200.0 : True在这种情况下，约束的 Body 为 0，Lower 和 Upper 都为 200，相当于一个恒成立的约束。
使用等式形式的元组表示法时需要格外小心：
from pyomo.environ import ConcreteModel, Constraint model = ConcreteModel() model.Cons3 = Constraint(expr=(0, 200)) model.Cons3.pprint()
输出结果如下：
Cons3 : Size=1, Index=None, Active=True
Key : Lower : Body : Upper : Active
None : 0.0 : 200.0 : 0.0 : TruePyomo 无法确定哪个值应该是约束体，哪个值应该是右侧。虽然可以通过 .expr.arg(0) 和 .expr.arg(1) 访问这些值，但最好避免使用这种模糊的表示法。
总结
本文介绍了在 Pyomo 中逐步构建约束的方法，并讨论了使用 Expression 组件、替换约束表达式以及创建 "空" 约束的注意事项。通过掌握这些技巧，您可以更加灵活地构建复杂的优化模型。请记住，Pyomo 的表达式具有不可变性，因此需要采用一些特殊的技巧才能实现动态修改约束的目的。
