如何用Python API为Gurobi定义依赖且类型不同的多索引变量？

原代码的核心问题 #

1. 变量索引不匹配：
   原代码用itertools.product(I.index, L)生成变量时，L的元素是(k,j)元组，导致变量x_ikj的索引结构为(i, (k,j))（例如('i0', ('k0','j4'))），但约束中却尝试用x_ikj[i, k, j]调用三元组索引(i,k,j)，Gurobi无法匹配到对应变量，触发错误。

2. 空集合约束的可行性问题：
   N_i['i2']是空列表，对应约束会变成0 ≥ 31，属于不可行约束，需要确认数据是否正确，或调整theta['i2']的取值。

3. 缺少目标函数定义：
   优化问题必须明确目标函数（最小化/最大化某一表达式），原代码未定义，模型无法完成求解。

修正后的完整代码 #

import gurobipy as gp
from gurobipy import GRB
import pandas as pd

# 定义原始集合
I = pd.DataFrame(index=['i0', 'i1', 'i2', 'i3', 'i4'])
L = [('k0', 'j2'), ('k0', 'j3'), ('k0', 'j4'), ('k1', 'j3'), ('k2', 'j1'), ('k2', 'j2'), ('k2', 'j4')]
N_i = {
    'i0': [('k0', 'j4')],
    'i1': [('k0', 'j3'), ('k1', 'j3')],
    'i2': [],
    'i3': [('k0', 'j2'), ('k0', 'j3'), ('k0', 'j4'), ('k2', 'j2'), ('k2', 'j4')],
    'i4': [('k0', 'j4')]
}
theta = {'i0': 27, 'i1': 36, 'i2': 31, 'i3': 14, 'i4': 41}

# 创建模型
model1 = gp.Model('Location-allocation problem 1')

# 修正变量定义：用(i,k,j)三元组作为索引，仅创建N_i中存在的有效组合（提升效率）
valid_indices = [(i, k, j) for i in I.index for (k,j) in N_i[i]]
x_ikj = model1.addVars(valid_indices, vtype=GRB.INTEGER, name='x_ikj')

# 定义目标函数（示例：最小化所有x_ikj的和，需根据实际需求修改）
model1.setObjective(gp.quicksum(x_ikj[i,k,j] for i in I.index for (k,j) in N_i[i]), GRB.MINIMIZE)

# 添加约束：跳过空集合的不可行约束
for i in I.index:
    if N_i[i]:
        model1.addConstr(
            gp.quicksum(x_ikj[i,k,j] for (k,j) in N_i[i]) >= theta[i],
            name=f'Constraint_{i}'
        )
    else:
        print(f"警告：i={i}的N_i为空，约束0≥{theta[i]}不可行，已跳过该约束")

# 求解模型
model1.optimize()

# 输出最优解（如果求解成功）
if model1.status == GRB.OPTIMAL:
    print("最优解：")
    for var in x_ikj:
        if x_ikj[var].X > 1e-6:
            print(f"{x_ikj[var].VarName} = {x_ikj[var].X}")

关键修正点说明 #

1. 变量索引对齐：
   将变量索引改为(i,k,j)三元组，与约束中的调用方式完全匹配，解决索引不匹配问题。

2. 目标函数补充：
   示例中添加了最小化变量总和的目标函数，你需要根据实际业务需求（如最小化成本、最大化覆盖范围等）修改目标表达式。

3. 空集合约束处理：
   对N_i[i]为空的情况增加判断，避免生成不可行约束。如果业务上i2必须满足约束，需检查数据合理性（如调整theta['i2']为0，或补充N_i['i2']的元素）。

4. 变量创建效率：
   仅创建N_i中存在的变量组合，减少模型规模，提升求解速度。

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