Pyomo优化模式下访问前序变量及Gurobi代码转Pyomo方法

• 若要获取求解后上一步的变量数值：完成模型求解后，直接通过变量对象的.value属性访问。例如，若变量是带时间索引的model.A[t]，求解后model.A[t-1].value即为上一步的计算值。
• 若在建模阶段需要引用前序步的变量（如时间序列约束）：提前定义带步长索引的变量集合，通过索引直接调用前序步变量即可，无需额外存储，示例见下方代码中的约束定义。

Pyomo推荐通过索引集合管理变量，替代手动拼接键名的字典方式，以下是等效实现：

from pyomo.environ import ConcreteModel, Var, Constraint, NonNegativeReals

# 创建Pyomo模型实例
model = ConcreteModel()

# 定义时间步索引：覆盖0到76（对应原代码的初始步0和循环步1-76）
model.time_steps = range(77)

# 定义变量，设置初始值、类型和上下限
model.A = Var(model.time_steps, within=NonNegativeReals, initialize={0: 1})
model.B = Var(model.time_steps, within=NonNegativeReals, bounds=(0, 77), initialize={0: 2})
model.C = Var(model.time_steps, within=NonNegativeReals, initialize={0: 3})

# 定义滞后约束规则
def lag_constraint(model, t):
    if t == 0:
        return Constraint.Skip  # 初始步无前置变量，跳过约束
    return model.C[t] <= model.A[t] * model.B[t-1]

# 为所有时间步添加约束
model.lag_constraints = Constraint(model.time_steps, rule=lag_constraint)

代码说明 #

• 用model.time_steps索引集合替代原循环的ix，更贴合Pyomo的结构化建模思路
• 变量的初始值通过initialize参数传入，对应原代码中variables字典的初始值
• 变量的非负约束、上下限通过within和bounds参数设置，对应Gurobi的lb、ub、vtype
• 约束通过规则函数批量生成，自动遍历所有时间步，仅对t>=1的步添加滞后约束

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