如何用多维Pyomo Set正确索引列表初始化的Pyomo参数

问题根因 #

报错核心是Param初始化的传参逻辑不符合Pyomo的解析规则：

• 写法initialize = paramListInitialize(M)会提前执行初始化函数，把返回的普通列表直接传入initialize参数。
• Pyomo接收到普通列表作为初始化输入时，默认以列表的整数下标（0、1、2……）为索引匹配参数位置，不会主动对齐传入的多维Set的元组索引。
• 由于目标Param以二元元组为合法索引，不存在整数0这个索引项，直接触发KeyError。
• 一维Set场景下列表初始化能运行属于巧合：一维Set的成员本身就是0、1、2类整数，刚好和列表下标匹配，不是通用的正确逻辑。

正确初始化方式 #

多维索引的Param不要直接用普通列表做初始化输入，选择以下任意一种规范写法即可：

方式1：返回索引-值映射字典 #

把初始化函数的返回值从列表改为「元组索引:参数值」格式的字典，传参时只传入函数对象本身（不要加括号提前调用执行函数）：

## Testing pyomo sets and indexing
import pyomo.environ as pyo
import numpy as np
import itertools as iter

M = pyo.ConcreteModel()
a = np.arange(3)
b = np.arange(3)
T = list(iter.product(a,b))
M.T = pyo.Set(initialize = T)

def paramDictInitialize(M):
    param_map = {}
    for idx in M.T:
        # 此处替换为实际需要赋值的参数值，示例暂用索引本身作为值
        param_map[idx] = idx
    return param_map

# 注意：不要给函数加(M)，直接传函数名即可
M.Param3 = pyo.Param(M.T, initialize=paramDictInitialize)

# 验证赋值结果
M.Param3.pprint()

方式2：单索引逐值返回 #

让初始化函数接收「模型+单个索引」两个入参，直接返回当前索引对应的参数值，Pyomo会自动遍历索引集完成全量赋值：

def paramSingleInit(M, idx):
    # idx为M.T遍历过程中的每一个二元元组，例如(0,0)、(0,1)
    # 此处替换为实际需要赋值的参数值
    return idx

M.Param3 = pyo.Param(M.T, initialize=paramSingleInit)

补充说明 #

• 现有多维Set的定义逻辑完全正确，不需要调整Set的构建方式，问题仅出在Param的初始化传参环节。
• 哪怕是一维索引场景，也不建议用普通列表按位置初始化Param，避免后续索引顺序调整导致赋值错位，优先使用上述两种规范写法。

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