Pyomo动态初始化集合问题：如何迭代创建Set组件适配用户输入

原有静态初始化方式 #

当资产数量固定时，通常直接逐个定义Set组件：

import pyomo.environ as pyo

asset_id = {}
asset_id[0] = ["A1", "A2", "A3", "A4"]
asset_id[1] = ["B1", "B2", "B3", "B4"]

model = pyo.ConcreteModel()
model.asset1_id = pyo.Set(initialize=asset_id[0])
model.asset2_id = pyo.Set(initialize=asset_id[1])

错误的动态创建尝试 #

当资产数量由用户输入决定（数量未知）时，若直接通过字典赋值创建Set组件会触发报错：

model = pyo.ConcreteModel()
model.asset = {}
for i in range(0, 2):
    model.asset[i] = pyo.Set(initialize=asset_id[i])

问题在于Pyomo的ConcreteModel对象不支持通过字典属性直接管理组件，这种方式无法让Pyomo识别并追踪这些Set组件。

可行的实现方法 #

方法1：使用add_component动态添加组件

这是Pyomo官方推荐的动态添加组件的方式，通过model.add_component()可以灵活创建并注册Set组件：

import pyomo.environ as pyo

# 模拟用户输入生成的资产集合字典，实际可替换为用户输入逻辑
asset_id = {
    0: ["A1", "A2", "A3", "A4"],
    1: ["B1", "B2", "B3", "B4"],
    # 可根据用户输入扩展更多条目
}

model = pyo.ConcreteModel()

for idx in asset_id:
    # 自定义组件名称，比如asset_0_id、asset_1_id
    component_name = f"asset_{idx}_id"
    # 动态添加Set组件
    model.add_component(component_name, pyo.Set(initialize=asset_id[idx]))

# 打印模型验证组件是否创建成功
model.pprint()

方法2：使用索引集合（Indexed Set）统一管理

如果所有资产集合属于同一逻辑类别，使用Indexed Set更符合Pyomo的建模规范，便于统一管理：

import pyomo.environ as pyo

# 根据用户输入生成资产类型索引集合，这里模拟2类资产
asset_type_count = 2  # 实际由用户输入决定
model = pyo.ConcreteModel()
model.asset_types = pyo.RangeSet(0, asset_type_count - 1)

# 资产集合字典
asset_id = {
    0: ["A1", "A2", "A3", "A4"],
    1: ["B1", "B2", "B3", "B4"],
}

# 定义索引Set，每个索引对应一个资产集合
model.asset_id = pyo.Set(model.asset_types, initialize=asset_id)

# 验证组件
model.pprint()

内容的提问来源于stack exchange，提问作者cyntha