提问者:小点点

求线性pyomo约束的系数


我想获得pyomo模型m的线性约束c的系数。

例如,对于

    m= ConcreteModel()
    m.x_1 = Var()
    m.x_2 = Var()
    m.x_3 = Var(within = Integers)
    m.x_4 = Var(within = Integers)
    m.c= Constraint(expr=2*m.x_1 + 5*m.x_2 + m.x_4 <= 2)

我想得到数组c_coef=[2,5,0,1]

这个问题的答案解释了如何获得线性约束中出现的所有变量,我可以轻松地使用它为约束中没有出现的变量创建零系数。然而,我正在和非零系数作斗争。我当前的方法使用私有属性\u coef,即c\u nzcoef=m.c.body_coef我可能不应该使用它。

获得非零系数的正确方法是什么?


共1个答案

匿名用户

获取线性表达式系数的最简单方法是使用“规范表示”数据结构:

from pyomo.repn import generate_canonical_repn
# verify that the expression is linear
if m.c.body.polynominal_degree() == 1:
    repn = generate_canonical_repn(m.c.body)
    for i, coefficient in enumerate(repn.linear or []):
        var = repn.variables[i]

这应该适用于从4.0到至少5.3的任何Pyomo版本。