在Pyomo或Gurobi中具有不同索引长度的多维变量

时间:2018-12-15 17:35:41

标签: python optimization gurobi pyomo

我想解决Python中的优化问题。我正在尝试定义变量x_ {g,h},其中索引g属于集合G,索引h属于集合H(g),即,索引h的集合对于不同的索引g有所不同。有什么方法可以在Pyomo或Gurobi-Python中用这些索引定义变量x?

在Pyomo中,我试图像一个循环那样定义它,

for g in p.keys():
    for h in range(0,p.rop[g].npairs,1):
        model.x_gen_h = Var(g,h,within=NonNegativeReals)

我收到此错误:

TypeError: 'int' object is not iterable.

感谢您的帮助或评论!

2 个答案:

答案 0 :(得分:2)

诀窍在于定义用于索引变量的索引集。 Pyomo不支持循环遍历单个索引并将它们一次添加到一个Var中。您应该使用一些聪明的Python代码来构建整个索引集。例如,您可以使用类似这样的方法来过滤出所需的索引:

m = ConcreteModel()

m.g = Set(initialize=[1,2,3])

h = {1:['a','b'], 2:['b','c'], 3:['c','d']}
m.h_all = Set(initialize=set(sum(h.values(),[]))) # Extract unique h values

# Initialize set to be entire cross product of g and h and then filter desired values
m.hg = Set(initialize=m.g*m.h_all, filter=lambda m,g,hi:hi in h[g])
m.x = Var(m.hg, within=NonNegativeReals)

一个更好的选择是:

h = {1:['a','b'], 2:['b','c'], 3:['c','d']}
m.hg = Set(initialize=list((i,j) for i in h.keys() for j in h[i])

答案 1 :(得分:1)

我将看一下Pyomo文档https://pyomo.readthedocs.io/en/latest/tutorial_examples.html中引用的一些示例模型。

您不需要使用for循环来构造变量。