如何在远程Jupyter Notebook中为Julia脚本传递参数

问题根源 #

你直接在jupyter nbconvert命令后追加参数的方式是错误的：

1. jupyter nbconvert的位置参数默认被识别为待转换的Notebook文件路径，所以arg1/arg2/arg3会被当作不存在的文件，触发匹配失败的警告。
2. 在Jupyter环境中，Julia的ARGS变量存储的是Jupyter内核启动时的系统参数（比如临时配置文件路径），并非你传递的自定义参数，因此才会输出/tmp/tmp8vuj5f79.json这类内容。

正确解决方案 #

方法1：使用nbconvert的--ExecutePreprocessor.args参数传递 #

这是官方推荐的给执行中的Notebook传递参数的方式：

1. 执行命令时通过该选项传递参数：

jupyter nbconvert --to notebook --execute my_notebook.ipynb --ExecutePreprocessor.args="10 20 30"

2. 在Notebook的Julia代码中，通过IJulia.args获取参数（需确保已安装IJulia包）：

using IJulia
# 获取传递的参数
args_list = IJulia.args
# 解析为整数
x1 = parse(Int16, args_list[1])
x2 = parse(Int16, args_list[2])
x3 = parse(Int16, args_list[3])

println("参数值：", x1, ", ", x2, ", ", x3)

方法2：通过环境变量传递参数 #

如果不想依赖IJulia的API，也可以用环境变量传递：

1. 执行命令前设置环境变量：

ARG1=10 ARG2=20 ARG3=30 jupyter nbconvert --to notebook --execute my_notebook.ipynb

2. 在Julia代码中读取环境变量：

x1 = parse(Int16, ENV["ARG1"])
x2 = parse(Int16, ENV["ARG2"])
x3 = parse(Int16, ENV["ARG3"])

println("参数值：", x1, ", ", x2, ", ", x3)

方法3：通过外部配置文件传递（适合参数较多的场景） #

1. 编写一个JSON配置文件params.json：

{"x1":10, "x2":20, "x3":30}

2. 执行命令时指定配置文件路径：

PARAM_FILE="./params.json" jupyter nbconvert --to notebook --execute my_notebook.ipynb

3. 在Julia代码中读取配置文件：

using JSON
# 读取环境变量中的配置文件路径
param_path = ENV["PARAM_FILE"]
params = JSON.parsefile(param_path)

x1 = parse(Int16, params["x1"])
x2 = parse(Int16, params["x2"])
x3 = parse(Int16, params["x3"])

println("参数值：", x1, ", ", x2, ", ", x3)

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