我正在努力在Python Pyomo中使用for循环制作多个约束。我的代码概念如下

T = 504
model.times = RangeSet(0, T-1)
model.Machines = Set(initialize = Machines_parameters.keys())

然后我划分了模型。在3个相同长度的集合中执行时间，现在我想为集合的每个部分编写特定的约束。以下是代码的等效简化版本：

for k in range(3): #number of parts in which I've divided the set [(0,167),(168,335),(336,503)]
    for j in range(cluster_index[k][0], cluster_index[k][1]):   #respectively first and last number of the subset
        def logic_constr(model,i):
            if j >= const1[k] and j < const2[k]:
                return model.z[i, j + 1] - model.z[i, j] == model.dsu[i, j + 1] - model.dsd[i, j + 1]
            else j==const2[k]:
                return model.z[i,const2[k]] - model.z[i,j] == model.dsu[i,const1[k]] - model.dsd[i,const1[k]]
        model.logic_constr = Constraint(model.Machines, rule = logic_constr)

我想做的是迭代创建504个不同的约束，每个约束都有自己的规则。你对怎么做有什么建议吗？