python - 创建具有递增整数的变量

Question

我想创建一个包含 1000 个基本 pyomo 变量的 pyomo 模型。我知道像下面的脚本那样这样做是一个坏主意。而且它也不起作用。我希望您理解这个想法并能够帮助我。

import pyomo.core as pyomo

def create_model_a():
    m = pyomo.ConcreteModel()

    for i in range(1000):
        m.var_i = pyomo.Var(within=pyomo.NonNegativeReals)

    return m

所以基本上我没有将 m.var_0 = ... 写入 m.var_999 = ...，而是使用了 for 循环，当然这样它就可以了不起作用，但想法是创建 1000 个变量，而无需硬编码 m.var_0、m.var_1、m.var_2 等直到m.var_999。 我该怎么做？

我想创建这个不是为了建模任何东西，但我想在这个函数上使用内存配置文件来了解具有 1000 个变量的 pyomo 模型需要多少内存。

Ps:我尝试了以下操作，但它不起作用(当我转换 m.pprint() 时看不到任何声明):

def create_model_a():
    m = pyomo.ConcreteModel()
    m.var = {}
    for i in range(1000):
        m.var[i] = pyomo.Var(within=pyomo.NonNegativeReals)

    return m

PS2:也检查了 How to increment variable names/Is this a bad idea和 How do I create a variable number of variables? ...遗憾的是没有帮助

Answer 1

如果你真的想了解拥有许多 Pyomo 变量对内存的影响，你应该将拥有许多单例变量的情况与拥有一个大索引变量的情况进行比较。两者的示例如下:

# Make a large indexed variable
m = ConcreteModel()
m.s = RangeSet(1000)
m.v = Var(m.s, within=NonNegativeReals)

# Make many singleton variables
m = ConcreteModel()
for i in range(1000):
    name = 'var_' + str(i)
    m.add_component(name, Var(within=NonNegativeReals))