微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

为每次运行在 Julia 中创建新目录

如何解决为每次运行在 Julia 中创建新目录

我正在运行生成绘图和文本文件的 Julia 代码。在相关代码所在的同一文件夹中存在一个输出文件夹。对于第一次运行,我创建了一个“Run_1”文件夹,其中包含“Plots”和“Data”子文件夹:

fig_path = @__DIR__
mkdir(fig_path*"/Output/Run_1/")
mkdir(fig_path*"/Output/Run_1/Plots/")
mkdir(fig_path*"/Output/Run_1/Data/")

绘图后,我将图形保存到“绘图”:

fig_name = "test_figure"
savefig(fig_path*"/Output/Run_1/Plots/"*fig_name*".pdf")

并将输出文件(包含在“output_matrix”中)保存到“Data”:

outfile_1 = fig_path*"/Output/Run_1/Data/test_data.txt"
open(outfile_1,"w") do f1    
    writedlm(f1,output_matrix)    
end

但是,我想多次运行此代码。每次运行时,它应该在“输出文件夹中创建一个新的“运行”文件夹,即第一次运行它的 Run_1,第二次运行它的 Run_2,依此类推。以前运行的所有文件夹都不会被删除。在每个 Run 文件夹中,有一个“Plots”和一个“Data”文件夹,我在每次运行时将绘图和数据保存到它们各自的文件夹中。我怎样才能让 Julia 以这种方式更新文件名?

解决方法

ispath 函数检查文件系统中是否存在文件或目录。

如果您想保持像 Run_1...Run_N 这样的命名约定,这样的事情可能会有所帮助:

function mk_output_dir()
    i = 1
    while true
        dir_name = joinpath(@__DIR__,"Output","run_$i")
        if !ispath(dir_name)
            mkpath(dir_name)
            return dir_name
        end
        i += 1
    end
end

这会产生:

# First run
julia> top_dir = mk_output_dir()
"/tmp/Output/run_1"

julia> mkdir(joinpath(top_dir,"Plots"))
"/tmp/Output/run_1/Plots"

julia> mkdir(joinpath(top_dir,"Data"))
"/tmp/Output/run_1/Data"
# Second run
julia> top_dir = mk_output_dir()
"/tmp/Output/run_2"

julia> mkdir(joinpath(top_dir,"Plots"))
"/tmp/Output/run_2/Plots"

julia> mkdir(joinpath(top_dir,"Data"))
"/tmp/Output/run_2/Data"

请注意,如果您同时启动程序的两个实例,可能会发生竞争条件。



或者,我个人倾向于在创建这样的目录结构时使用涉及时间戳的命名约定。这是一个最小的例子:

using Dates
function mk_output_dir()
    timestamp = Dates.format(now(),"YYYYmmdd-HHMMSS")
    dir_name = joinpath(@__DIR__,"run_$timestamp")
    @assert !ispath(dir_name) "Somebody else already created the directory"
    mkpath(dir_name)
    return dir_name
end

产生这样的东西:

julia> top_dir = mk_output_dir()
"/tmp/Output/run_20201229-210835"

julia> mkdir(joinpath(top_dir,"Plots"))
"/tmp/Output/run_20201229-210835/Plots"

julia> mkdir(joinpath(top_dir,"Data"))
"/tmp/Output/run_20201229-210835/Data"
,

也许是这样的:

function mkresultdir(fig_path)
    for i=1:1000
        rundir = joinpath(fig_path,"run_$i")
        if !isdir(rundir)
            mkdir(rundir)
            return rundir
        end
     end
     error("too many results on disk,time for a cleanup!")
 end

res_dir_1 = mkresultdir("/home/my_user/results")
res_dir_2 = mkresultdir("/home/my_user/results")

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。