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

我会带着你远行 2023-09-30 08:39 115阅读 0赞

[parser 官方文档][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()用法——命令行选项、参数和子命令解析器][python_parser.add_argument]

[parser]: https://docs.python.org/zh-cn/3/library/argparse.html#argumentparser-objects
[python_parser.add_argument]: https://blog.csdn.net/qq_34243930/article/details/106517985