英文:
Dynamic variables and constraints in Pyomo
问题
You can modify your Pyomo code to dynamically create the variables and constraints based on the input 'N'. Here's the adjusted code:
!pip install -q pyomo!apt-get install -y -qq glpk-utilsimport pyomo.environ as pyofrom pyomo.environ import *from pyomo.opt import SolverFactoryalpha = 0.8N = 9 # This is the input that can be changed and will impact the number of variables and constraintsmodel = pyo.ConcreteModel()model.a = pyo.Var(domain=NonNegativeReals)# Create a dictionary to hold the variables 'b' dynamicallymodel.b = pyo.Var(range(1, N+1), domain=NonNegativeReals)# Define the objective functionTTL = model.a + (1/(1-alpha)) * sum(model.b[i] for i in range(1, N+1))# Create constraints dynamicallymodel.c = ConstraintList()for i in range(1, N+1):model.c.add(expr = model.b[i] + model.a >= 0)model.obj = pyo.Objective(expr= TTL, sense=minimize)
With this code, you can change the value of 'N,' and the code will dynamically create the corresponding variables and constraints.
英文:
I need to solve an LP problem in Pyomo. The number of variables and constraints changes depending on the purpose it will be used and can be freely changed by the user (input 'N' in the code below). I did a code for a fixed number of variables and constraints (N=9). It is working well but I don't know how to make it dynamically create the variables and constraints as the input N changes. The code below is a simplified version of my code for better understanding.
!pip install -q pyomo!apt-get install -y -qq glpk-utilsimport pyomo.environ as pyofrom pyomo.environ import *from pyomo.opt import SolverFactoryalpha = 0.8N = 9 # this is the input that can be changed and will impact on the number of variables and constraintsmodel = pyo.ConcreteModel()model.a = pyo.Var(domain=NonNegativeReals)# the number of the variables below must follow the input 'N' that in this case is 9model.b1 = pyo.Var(domain=NonNegativeReals)model.b2 = pyo.Var(domain=NonNegativeReals)model.b3 = pyo.Var(domain=NonNegativeReals)model.b4 = pyo.Var(domain=NonNegativeReals)model.b5 = pyo.Var(domain=NonNegativeReals)model.b6 = pyo.Var(domain=NonNegativeReals)model.b7 = pyo.Var(domain=NonNegativeReals)model.b8 = pyo.Var(domain=NonNegativeReals)model.b9 = pyo.Var(domain=NonNegativeReals)# below is a calculation during the building of obj function that also needs to change based on input 'N'. Necessary to sum all variables 'b' createdTTL = model.a + (1/(1-alpha)) * (model.b1 + model.b2 + model.b3 + model.b4 + model.b5 + model.b6 + model.b7 + model.b8 + model.b9)# below are the constraints that should change based on the input 'N' as wellmodel.c1 = Constraint(expr = model.b1 + model.a >= 0)model.c2 = Constraint(expr = model.b2 + model.a >= 0)model.c3 = Constraint(expr = model.b3 + model.a >= 0)model.c4 = Constraint(expr = model.b4 + model.a >= 0)model.c5 = Constraint(expr = model.b5 + model.a >= 0)model.c6 = Constraint(expr = model.b6 + model.a >= 0)model.c7 = Constraint(expr = model.b7 + model.a >= 0)model.c8 = Constraint(expr = model.b8 + model.a >= 0)model.c9 = Constraint(expr = model.b9 + model.a >= 0)model.obj = pyo.Objective(expr= TTL, sense=minimize)
For the creation of the variables, I am using the code below, which I understand may work. But once I cannot run the entire code with this variable declaration, I even don't know if it really works.
Using this
model.b = pyo.Var(range(N), domain=NonNegativeReals)
Instead of this
model.b1 = pyo.Var(domain=NonNegativeReals)model.b2 = pyo.Var(domain=NonNegativeReals)model.b3 = pyo.Var(domain=NonNegativeReals)model.b4 = pyo.Var(domain=NonNegativeReals)model.b5 = pyo.Var(domain=NonNegativeReals)model.b6 = pyo.Var(domain=NonNegativeReals)model.b7 = pyo.Var(domain=NonNegativeReals)model.b8 = pyo.Var(domain=NonNegativeReals)model.b9 = pyo.Var(domain=NonNegativeReals)
How can I adjust the code above to meet the application requirements?
Many thanks in advance!
答案1
得分: 0
以下是翻译的代码部分：
import pyomo.environ as pyovar_count = int(input("有多少个变量："))m = pyo.ConcreteModel()m.N = pyo.Set(initialize=range(var_count))m.x = pyo.Var(m.N)m.pprint()
希望这对您有所帮助。如果有任何其他翻译需求，请随时告诉我。
英文:
Instead of making the variables 1-by-1, which is difficult to build and almost impossible to manage, just make an index set from whatever "dynamic" stuff happens in your model and use that. It is much more robust and succinct. A simple example... Main point, dynamically make the sets and then just use them to index your variables.
Code:
import pyomo.environ as pyovar_count = int(input("how many vars: "))m = pyo.ConcreteModel()m.N = pyo.Set(initialize=range(var_count))m.x = pyo.Var(m.N)m.pprint()
Sample run...
how many vars: 41 Set DeclarationsN : Size=1, Index=None, Ordered=InsertionKey : Dimen : Domain : Size : MembersNone : 1 : Any : 4 : {0, 1, 2, 3}1 Var Declarationsx : Size=4, Index=NKey : Lower : Value : Upper : Fixed : Stale : Domain0 : None : None : None : False : True : Reals1 : None : None : None : False : True : Reals2 : None : None : None : False : True : Reals3 : None : None : None : False : True : Reals
通过集体智慧和协作来改善编程学习和解决问题的方式。致力于成为全球开发者共同参与的知识库，让每个人都能够通过互相帮助和分享经验来进步。
评论