Pyomo参数声明报错：为何无法用NumPy数组初始化？

问题重现 #

尝试用NumPy数组初始化Pyomo二维参数时触发KeyError: "Index '0' is not valid for indexed component 'c'"错误，原代码如下：

model = pyo.ConcreteModel()

V = range(20)

model.V = pyo.Set(initialize = V, doc = 'Set: clients and depots', within = pyo.NonNegativeIntegers)

c = np.zeros((len(V), len(V)))

model.c = pyo.Param(V, V, initialize = c, doc = 'Param: distances', within = pyo.NonNegativeReals)

原因解析 #

Pyomo的Param组件对多维索引的初始化逻辑有明确要求：必须传入键为索引元组、值为对应参数值的字典，而非NumPy数组。

Pyomo的集合（比如这里的V）是独立的成员集合，即便集合元素是连续整数，Pyomo也会将其当作离散的索引项处理，而非数组下标。当直接传入NumPy数组时，Pyomo会尝试用集合元素（如0、1等）匹配数组的访问逻辑，但NumPy数组的索引方式和Pyomo期望的键值映射逻辑不兼容——Pyomo需要明确的(i,j)元组定位每个参数值，数组无法直接提供这种映射关系，最终导致索引匹配失败，抛出KeyError。

正确实现方式 #

将NumPy数组转换成Pyomo可识别的字典结构即可解决问题，代码如下：

c_dict = {(i, j): c[i][j] for i in V for j in V}
model.c = pyo.Param(V, V, initialize=c_dict, doc='Param: distances', within=pyo.NonNegativeReals)

这段代码通过字典推导式，把NumPy数组中每个位置(i,j)的元素值，映射为字典的键值对(i,j): c[i][j]，让Pyomo可以精准匹配每个索引对应的参数值。

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