如何在pyomo中同时放置多个约束？

Question

我想一次输入许多约束。我的约束如下。

def ss1_rule(model):
    return model.ss[1,1] - model.sss[1,1] <= 0.0
model.ss1 = Constraint(rule=ss1_rule)

def ss2_rule(model):
    return model.ss[1,1] - model.sss[2,1] <= 0.0
model.ss2 = Constraint(rule=ss2_rule)

def ss3_rule(model):
    return model.ss[1,1] - model.sss[3,1] <= 0.0
model.ss3 = Constraint(rule=ss3_rule)

有没有办法一次输入常规约束？

谢谢。

Answer 1

有多种方法可以实现这一点。您希望使用Set或RangeSet。

model.my_set = Set(initialize=[1, 2, 3])
def ss_rule(model, s):
    return model.ss[1, 1] = model.sss[s, 1] <= 0.0
model.ss = Constraint(model.my_set, rule=ss_rule)

或使用快捷方式表示法的等价物：

model.my_set = Set(initialize=[1, 2, 3])
@model.Constraint(model.my_set)
def ss_rule(model, s):
    return model.ss[1, 1] = model.sss[s, 1] <= 0.0