Jupyter Notebook运行Python代码实现传参方式

argv[0]: ‘路径\test.py' （源代码文件名）
argv[1]: ‘-f'
argv[2]: '路径\kernel-…json'

python test.py test.txt
argv[1]为 test.txt

#需要sys模块来使用argv
from sys import argv
#查看当前的argv列表
print(len(argv))
print(argv)
#直接修改argv
argv[1] = 'test.txt'
#也可以添加新的参数
argv.append('test.txt')

# autorun.py
import os
# print(os.getcwd())
over = [  # 之前手工改参数跑完的参数组合
    [0, 1, 1], [0, 1, 2], [0, 1, 3],
    [0, 2, 1],
    [1, 0, 1],
    [1, 2, 1]
]
for alpha in range(1, 4, 1):
    for beta in range(3):
        for gamma in range(3):
            if [alpha, beta, gamma] in over:
                continue
            os.system(f'python main.py --alpha {alpha} --beta {beta} --gamma {gamma}')

# test.py
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--number', type=int, default=0, help='number')
parser.add_argument('--string', type=str, default='abc', help='string')
args = parser.parse_args()
print('number:', args.number, type(args.number))
print('string:', args.string, type(args.string))

# test.ipynb
for i in range(3):
    for s in ('a', 'b', 'c'):
        %run test.py --number $i --string $s