如何在Python的Pyomo模型中实现变量矩阵与常量矩阵相乘约束？

1. 初始化Pyomo模型与变量 #

首先导入Pyomo核心模块，创建模型实例，并定义1-based的行、列索引（贴合数学场景的索引方式）：

from pyomo.environ import ConcreteModel, Var, RangeSet, Constraint, Reals

# 创建模型
model = ConcreteModel()

# 定义行（1-21）和列（1-201）的索引集合
model.row_indices = RangeSet(1, 21)
model.col_indices = RangeSet(1, 201)

# 定义21×201的变量矩阵x，可根据需求调整domain（如NonNegativeReals）
model.x = Var(model.row_indices, model.col_indices, domain=Reals)

2. 实现矩阵约束逻辑 #

你的约束要求x[1:21, 2:11] = A × x[1:21, 1:10]，其中A是对角元全为10的21×21对角矩阵。由于对角矩阵的乘法特性，这个矩阵等式可以拆解为逐元素的线性约束：对于每一行i，第j列的x值等于10倍的第i行第j-1列的x值（j的范围是2到10，对应切片2:indx的1-based范围）。

直接通过约束规则遍历对应索引添加约束：

indx = 11

# 定义约束规则函数
def matrix_eq_constraint(model, i, j):
    return model.x[i, j] == 10 * model.x[i, j-1]

# 为指定行和列添加约束：行1-21，列2-10
model.matrix_constraint = Constraint(
    model.row_indices, 
    RangeSet(2, indx-1), 
    rule=matrix_eq_constraint
)

3. 关键说明 #

• 这里采用1-based索引是因为Pyomo的RangeSet默认支持，且更符合数学矩阵的表达习惯；如果你的代码使用0-based索引，只需将行调整为RangeSet(0,20)、列调整为RangeSet(0,200)，约束的列范围改为RangeSet(1, indx-1)（即1到9）即可。
• 对角矩阵A的乘法被简化为逐元素的缩放，避免了不必要的矩阵运算，既高效又符合Pyomo的约束定义方式。

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