切换到 Pyomo - 集合的语法问题 产量：

编程问答 2022-05-11

问题描述

我已经使用了代数建模语言 AMPL，但我现在正在改用 Python 和 Pyomo。

虽然我在语法上有点挣扎。在AMPL中，我会有这样的事情：

param M; 
param n{i in 0..M};
var b{k in 0..M-1,j in 1..n[k+1]};

如何在 Pyomo 中实现最后一行？

非常感谢任何帮助，谢谢！

最好的问候， 约翰内斯

解决方法

欢迎来到本站。

下面是一个我认为可以满足您要求的示例。在 pyomo 中构建稀疏集有多种 方法。您可以使用集合推导或其他方法在纯 python 中“在旁边”完成（如下例所示），或者您可以创建一个返回相同的规则。 documentation 中有一个不错的例子。

# sparse set example

import pyomo.environ as pyo

M = 4

mdl = pyo.ConcreteModel('sparse set example')

mdl.A =  pyo.Set(initialize=range(M))
sparse_index = {(k,j) for k in mdl.A for j in range(1,k+1)}    # just a little helper set-comprehension
mdl.LT = pyo.Set(within=mdl.A * mdl.A,initialize=sparse_index)  # "within" is optional...good for error checking

mdl.x =  pyo.Var(mdl.LT,domain=pyo.NonNegativeReals)

mdl.pprint()

产量：

3 Set Declarations
    A : Size=1,Index=None,Ordered=Insertion
        Key  : Dimen : Domain : Size : Members
        None :     1 :    Any :    4 : {0,1,2,3}
    LT : Size=1,Ordered=Insertion
        Key  : Dimen : Domain    : Size : Members
        None :     2 : LT_domain :    6 : {(2,1),(3,(1,3),(2,2),2)}
    LT_domain : Size=1,Ordered=True
        Key  : Dimen : Domain : Size : Members
        None :     2 :    A*A :   16 : {(0,0),(0,3)}

1 Var Declarations
    x : Size=6,Index=LT
        Key    : Lower : Value : Upper : Fixed : Stale : Domain
        (1,1) :     0 :  None :  None : False :  True : NonNegativeReals
        (2,2) :     0 :  None :  None : False :  True : NonNegativeReals
        (3,1) :     0 :  None :  None : False :  True : NonNegativeReals
        (3,3) :     0 :  None :  None : False :  True : NonNegativeReals

4 Declarations: A LT_domain LT x
amplpyomopython