Pyomo新手求助：如何从字典键定义cost(n)参数

嘿，作为Pyomo新手碰到这种参数映射的问题太正常了，我来一步步帮你搞定！

首先，你的思路方向是对的：先定义操作集合，再把字典里的score值映射成对应集合元素的参数。问题大概率出在参数初始化的方式上，下面给你两种靠谱的实现方法：

方法1：字典推导式初始化参数 #

先把字典里的score提取成Pyomo参数能识别的键值对字典，再传给Param：

from pyomo.environ import ConcreteModel, RangeSet, Param

# 你的示例操作字典
operations = {
    1: {"score": 100, "start_node": 2, "end_node": 3},
    2: {"score": 120, "start_node": 4, "end_node": 3},
    3: {"score": 155, "start_node": 5, "end_node": 2}
}

n_operations = len(operations)
model = ConcreteModel()

# 定义操作集合，和字典的整数键对应
model.O = RangeSet(1, n_operations)

# 提取每个操作对应的score，生成参数初始化字典
cost_init_data = {op: operations[op]['score'] for op in model.O}
# 定义参数cost，关联到集合O
model.cost = Param(model.O, initialize=cost_init_data)

方法2：用lambda函数直接初始化（更简洁） #

如果不想额外生成中间字典，可以直接在Param的initialize参数里用lambda函数读取字典值：

# 接上面的集合定义
model.cost = Param(model.O, initialize=lambda model, op: operations[op]['score'])

额外的灵活优化 #

如果你的operations字典的键不一定是连续整数（比如可能有缺失或者非整数键），可以直接用字典的键来生成集合，这样更稳妥：

# 直接用operations的键作为集合元素
model.O = Set(initialize=operations.keys())
# 后续参数定义和上面一样
model.cost = Param(model.O, initialize=lambda model, op: operations[op]['score'])

验证参数是否正确 #

你可以加一段代码验证参数值是否正确加载：

for op in model.O:
    print(f"cost({op}) = {model.cost[op].value}")

运行后应该会输出每个操作对应的score值，比如：

cost(1) = 100
cost(2) = 120
cost(3) = 155

之前你可能失败的原因：比如没有正确把集合元素和字典键关联，或者初始化时没有提取score键（直接传了整个operations字典），按照上面的方法调整后应该就能正常工作啦！

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