如何在Julia中为优化模型添加指定分段函数约束？

1. 读取Excel数据 #

首先使用XLSX.jl包读取Excel中的PR和HSE数据，这是Julia生态中处理Excel文件的常用工具。先安装依赖包：

using Pkg
Pkg.add(["XLSX", "JuMP", "GLPK"])

读取数据的示例代码：

using XLSX

# 定义你的Excel文件路径
excel_file = "your_data.xlsx"

pr = []
hse = []

XLSX.openxlsx(excel_file) do xf
    global pr, hse
    # 假设PR在Sheet1的A列（从第2行开始到第n_i+1行），HSE在B列（第2行到M+1行）
    pr = vec(XLSX.readdata(xf, "Sheet1", "A2:A10"))  # 根据实际范围调整
    hse = vec(XLSX.readdata(xf, "Sheet1", "B2:B5")) # 根据实际范围调整
end

n_i = length(pr)  # Ω集合的元素个数
M = length(hse)   # k的总数

2. 实现分段函数约束 #

根据分段条件的类型，分两种情况处理：

情况1：分段条件为确定性规则（如基于i/k的固定集合）

如果C_{i,k}的取值仅由i或k所属的固定集合决定（比如i属于Ω₁时取PR_i*HSE_k，属于Ω₂时取1），直接通过索引子集批量添加约束即可：

using JuMP, GLPK

# 创建优化模型（以GLPK求解器为例，可替换为Gurobi/CPLEX等）
model = Model(GLPK.Optimizer)

# 定义决策变量C[i,k]
@variable(model, C[1:n_i, 1:M])

# 定义分段集合（示例：前5个i属于Ω₁，其余属于Ω₂）
Ω₁ = 1:5
Ω₂ = 6:n_i

# 添加分段约束
@constraint(model, [i in Ω₁, k in 1:M], C[i,k] == pr[i] * hse[k])
@constraint(model, [i in Ω₂, k in 1:M], C[i,k] == 1)

# 添加目标函数（根据你的需求调整）
@objective(model, Min, sum(C[i,k] for i in 1:n_i, k in 1:M))

# 求解模型
optimize!(model)

# 查看结果
println("最优目标值: ", objective_value(model))
println("C的取值: ", value.(C))

情况2：分段条件依赖模型变量

如果分段规则依赖其他决策变量的取值（比如当某个变量x[i,k]满足特定条件时，C[i,k]取PR_i*HSE_k，否则取1），需要引入二进制变量建模逻辑选择：

using JuMP, GLPK

model = Model(GLPK.Optimizer)

# 定义决策变量
@variable(model, C[1:n_i, 1:M])
@variable(model, x[1:n_i, 1:M])  # 触发分段的变量
@variable(model, y[1:n_i, 1:M], Bin)  # 二进制变量：y=1时取乘积，y=0时取1

# 定义分段触发阈值（根据你的需求设置）
threshold = 0.5

# 约束C的分段取值
@constraint(model, [i in 1:n_i, k in 1:M], 
    C[i,k] == y[i,k] * pr[i] * hse[k] + (1 - y[i,k]) * 1)

# 用大M约束关联y和x的关系（M取足够大的正数）
M_big = 1e6
@constraint(model, [i in 1:n_i, k in 1:M], 
    x[i,k] >= threshold - M_big * (1 - y[i,k]))
@constraint(model, [i in 1:n_i, k in 1:M], 
    x[i,k] <= threshold + M_big * y[i,k])

# 添加目标函数和其他约束
@objective(model, Min, sum(C[i,k] for i in 1:n_i, k in 1:M))

optimize!(model)

3. 额外说明 #

• 如果C_{i,k}是已知参数而非决策变量，无需将其定义为模型变量，直接预计算数组即可：

  C = Matrix{Float64}(undef, n_i, M)
  for i in 1:n_i, k in 1:M
      # 替换为你的分段条件
      C[i,k] = i in Ω₁ ? pr[i] * hse[k] : 1
  end
  

• 确保Excel数据读取时，通过vec()将二维表格数据转为一维向量，避免索引错误。

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