使用IPOPT回调函数追加迭代信息到文件时仅写入少量异常数据求助

我正在处理一个优化问题，想通过回调函数把IPOPT每次迭代的相关信息写入文件。虽然能收集到所需信息，但写入文件时只出现了一些异常的迭代数据。以下是简化实现代码：

我设置了外层迭代循环（k）和内层IPOPT迭代循环：

thefile = "output.txt"      # 创建新文件
f = open(thefile, "w")      # 写入文件头
@printf(f,"%-10s %-10s %-10s\n ", "outer", "inner", "objval" )

k = 0
while k <= 100 
      iter = []
      objValVector = []

      function my_callback( alg_mod::Cint,
        iter_count::Cint,
        obj_value::Float64,
        inf_pr::Float64,
        inf_du::Float64,
        mu::Float64,
        d_norm::Float64,
        regularization_size::Float64,
        alpha_du::Float64,
        alpha_pr::Float64,
        ls_trials::Cint)                # 通过回调函数获取目标值
         append!(objValvector, obj_value)
         append!(iter, iter_count)
         return true
      end 
 MOI.set(model, Ipopt.CallbackFunction(), my_callback)
 optimize!(model); 

 f = open(thefile, "a");         # 以追加模式打开文件
 for i in 1:length(iter)
     @printf(f, "%-10s %-10s %-10s\n", 
            k, iter[i], objValvector[i])
 end

# 其他操作...
k += 1; 


end

生成的文件内容存在异常，仅写入了部分错误的迭代数据。

问题原因及修复方案 #

1. 变量拼写错误 #

你定义的数组是objValVector（大写V），但回调函数里写成了objValvector（小写v）。这会导致Julia自动创建一个新的全局变量，而非往你定义的局部数组中追加数据，内层循环的objValVector始终为空，最终写入的是全局变量的异常数据。

2. 文件资源未正确释放 #

每次打开文件后未调用close(f)，可能导致缓冲区数据未及时写入，甚至出现资源泄漏。建议改用open(...) do语法，自动管理文件生命周期。

3. 回调函数作用域混淆 #

在while循环内直接定义函数，容易出现作用域问题。可以用let块捕获当前循环的iter和objValVector，确保回调能正确访问局部变量。

修复后的代码 #

thefile = "output.txt"
# 写入文件头，用do语法自动关闭文件
open(thefile, "w") do f
    @printf(f,"%-10s %-10s %-10s\n", "outer", "inner", "objval")
end

k = 0
while k <= 100 
    iter = Int[]  # 指定数组类型，提升性能与稳定性
    objValVector = Float64[]

    # 用let块捕获当前循环的局部变量，避免作用域问题
    my_callback = let iter=iter, objValVector=objValVector
        function (alg_mod::Cint,
                  iter_count::Cint,
                  obj_value::Float64,
                  inf_pr::Float64,
                  inf_du::Float64,
                  mu::Float64,
                  d_norm::Float64,
                  regularization_size::Float64,
                  alpha_du::Float64,
                  alpha_pr::Float64,
                  ls_trials::Cint)
            push!(objValVector, obj_value)  # 单个元素追加用push!更高效
            push!(iter, iter_count)
            return true
        end
    end

    MOI.set(model, Ipopt.CallbackFunction(), my_callback)
    optimize!(model)

    # 追加写入数据，用do语法自动管理文件
    open(thefile, "a") do f
        for i in 1:length(iter)
            @printf(f, "%-10d %-10d %-10.6f\n", k, iter[i], objValVector[i])
        end
    end

    # 其他操作...
    k += 1
end

额外优化点 #

• 给数组指定具体类型（Int[]、Float64[]），避免类型不稳定，提升运行效率。
• 单个元素追加使用push!代替append!，更符合语义且性能更优。
• 格式化输出时指定类型（%d对应整数、%.6f对应浮点数），避免格式混乱。

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