Python：parser.add_argument()用法【命令行选项、参数、子命令解析器、sh脚本】

parser 官方文档

一、argparse使用

argparse 模块是 Python 内置的一个用于命令项选项与参数解析的模块，argparse 模块可以让人轻松编写用户友好的命令行接口。通过在程序中定义好我们需要的参数，然后 argparse 将会从 sys.argv 解析出这些参数。argparse 模块还会自动生成帮助和使用手册，并在用户给程序传入无效参数时报出错误信息。

main.py文件

import argparse
 
parser = argparse.ArgumentParser()
parser.add_argument('--seed', type=int, default=72, help='Random seed.')
parser.add_argument('--epochs', type=int, default=10000, help='Number of epochs to train.')
parser.add_argument('--use_multi_gpu', default=False, action='store_true', help='use multiple gpus')
 
args = parser.parse_args()
 
print(args.seed)
print(args.epochs)

打印内容如下：

/home/user/anaconda3/bin/python3.6 /home/user/lly/pyGAT-master/test.py
False
72
10000
 
Process finished with exit code 0

action的作用：训练时加上–use_multi_gpu参数就表示其值为store_true表示的True

二、sh脚本

main.sh文件：

python main.py \
--seed=$1 \
--epochs=$1

其中：

• $1：表示控制台输入的第一个参数；
• $2：表示控制台输入的第二个参数；

三、控制台输入

sh main.sh 20 500

---

---

---

参考资料：
python之parser.add_argument()用法——命令行选项、参数和子命令解析器