如何配置Gurobi变量使其从指定整数集合{0,1,4,16,64}取值？

这个需求完全可行，Gurobi提供了多种方式让变量只能从指定的离散整数集合中取值，以下是两种常用实现方法：

方法1：直接添加集合成员约束 #

利用Gurobi的GRB.INSET操作符，直接约束变量属于目标集合，代码简洁直观：

# 定义允许的取值集合
allowed_values = {0, 1, 4, 16, 64}

# 创建变量，无需设置上下界参数lb和ub
n_vars = model.addVars(gurobi_vars, name='n', vtype=GRB.INTEGER)

# 为每个变量添加集合成员约束
model.addConstrs(
    (n_vars[i] GRB.INSET allowed_values for i in gurobi_vars),
    name='n_in_allowed_set'
)

方法2：二进制变量线性化表示 #

如果你的建模场景要求使用线性约束（比如部分求解模块对非线性约束有限制），可以通过二进制变量来实现离散取值的选择：

values = [0, 1, 4, 16, 64]

# 为每个目标变量创建一组二进制选择变量
y = model.addVars(gurobi_vars, len(values), name='selection', vtype=GRB.BINARY)
# 创建目标变量集合
n_vars = model.addVars(gurobi_vars, name='n', vtype=GRB.INTEGER)

# 为每个变量添加约束：只能选中一个取值选项
for idx in gurobi_vars:
    # 确保每个变量对应二进制变量的和为1（唯一选择）
    model.addConstr(
        quicksum(y[idx, j] for j in range(len(values))) == 1,
        name=f'single_selection_{idx}'
    )
    # 将目标变量与选中的取值绑定
    model.addConstr(
        n_vars[idx] == quicksum(values[j] * y[idx, j] for j in range(len(values))),
        name=f'n_value_bind_{idx}'
    )

方法选择建议 #

• 方法1更简洁，适合绝大多数常规场景，Gurobi会自动处理集合约束的求解逻辑；
• 方法2适合需要严格线性模型的场景，比如和其他线性约束结合时兼容性更好。

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