argparse:Python命令行传参

简介: argparse:Python命令行传参

argparse模块(Python官方文档:argparse — 命令行选项、参数和子命令解析器 — Python 3.10.3 文档),可以用来在用命令行运行Python脚本时直接传参。

本文将首先给出直接可以套用的代码格式,然后在此基础上进行更多内容的介绍。


本文目前仅对argparse包的必要功能进行简要介绍,对其更深层次的应用可能会在后期继续更新。


argparse是Python自带的模块。

通过 import argparse 导入包。


一个简单的应用实例:使用argparse传入参数section_id和gpu_id,并获取两个参数(其中gpu_id直接转为torch.device1)


parser = argparse.ArgumentParser()
#创建解析器
#ArgumentParser 对象包含将命令行解析成 Python 数据类型所需的全部信息。
parser.add_argument("--section_id",default='1',type=str)
parser.add_argument("--gpu_id",default='0',type=str)
#添加参数。后文会对各入参有详细介绍
args = parser.parse_args()
#解析参数
arg_dict=args.__dict__  #将arg_dict转换为dict格式
section_id=arg_dict['section_id']
device = torch.device("cuda:"+arg_dict['gpu_id'] if torch.cuda.is_available() else "cpu")


在命令行运行如下语句即可:python example.py --section_id 2 --gpu_id 1(example.py为Python脚本,参数后面即为所需传入的值)

注意,如果使用anaconda作为环境管理器,如果想直接使用python命令,需要使terminal处在虚拟环境下;否则就要用Python解释器的路径代替python命令。



以下示例及整体内容架构参考自官方文档:


示例prog.py(获取一个列表,并计算其总和或最大值):


import argparse
parser = argparse.ArgumentParser(description='Process some integers.')
#description是在参数帮助文档之前显示的文本
parser.add_argument('integers', metavar='N', type=int, nargs='+',
                    help='an integer for the accumulator')
parser.add_argument('--sum', dest='accumulate', action='store_const',
                    const=sum, default=max,
                    help='sum the integers (default: find the max)')
#integers 属性将是一个包含一个或多个整数的列表
#accumulate 属性当命令行中指定了 --sum 参数时将是 sum() 函数,否则则是 max() 函数。
args = parser.parse_args()
print(args.accumulate(args.integers))


在命令行运行 python prog.py -h,得到提示信息:


usage: prog.py [-h] [--sum] N [N ...]
Process some integers.
positional arguments:
 N           an integer for the accumulator
options:
 -h, --help  show this help message and exit
 --sum       sum the integers (default: find the max)


计算最大值的示例:python prog.py 1 2 3 4


求和的示例:python prog.py 1 2 3 4 --sum


传入无效参数并报错的示例:python prog.py a b c

报错信息:


usage: prog.py [-h] [--sum] N [N ...]
prog.py: error: argument N: invalid int value: 'a'


add_argument() 方法(以下只列举一部分入参):

ArgumentParser.add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])


  1. name or flags - 一个命名或者一个选项字符串的列表,例如 foo 或 -f, --foo。

前面带-或--的就是optional argument,否则就是positional argument(在命令行中必须传入)。第一个传入的参数可以是一系列flags,也可以是单个argument name。

创建optional argument举例:parser.add_argument('-f', '--foo')(这是一种缩写形式,可以直接用-f或--foo传入参数)

创建positional argument举例:parser.add_argument('bar')

>>> parser = argparse.ArgumentParser(prog='PROG')
>>> parser.add_argument('-f', '--foo')
>>> parser.add_argument('bar')
>>> parser.parse_args(['BAR'])
Namespace(bar='BAR', foo=None)
>>> parser.parse_args(['BAR', '--foo', 'FOO'])
Namespace(bar='BAR', foo='FOO')
>>> parser.parse_args(['--foo', 'FOO'])
usage: PROG [-h] [-f FOO] bar
PROG: error: the following arguments are required: bar


  1. action - 当参数在命令行中出现时使用的动作基本类型。
  • 默认值 ‘store’:存储参数值
  • ‘store_const’:存储const keyword argument指定的值。
  • ‘store_true’:用参数传递布尔值。示例用法:


import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-a",action="store_true")
args = parser.parse_args()
print(type(args.a))
print(args.a)


不传入参数的输出:


<class 'bool'>
False


-a的输出:


<class 'bool'>
True


  1. nargs - 单个action关联几个参数。
  • ‘*’:整合所有参数为一个list
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', nargs='*')
>>> parser.add_argument('--bar', nargs='*')
>>> parser.add_argument('baz', nargs='*')
>>> parser.parse_args('a b --foo x y --bar 1 2'.split())
Namespace(bar=['1', '2'], baz=['a', 'b'], foo=['x', 'y'])


  •     ‘+’:与 ‘*’ 类似,但是如果没有传入至少一个参数就会报error。
>>> parser = argparse.ArgumentParser(prog='PROG')
#prog入参是程序名称,默认值为os.path.basename(sys.argv[0])
>>> parser.add_argument('foo', nargs='+')
>>> parser.parse_args(['a', 'b'])
Namespace(foo=['a', 'b'])
>>> parser.parse_args([])
usage: PROG [-h] foo [foo ...]
PROG: error: the following arguments are required: foo


  1. const - 被一些 action 和 nargs 选择所需求的常数。如果入参action='store_const’或’append_const’会将该值与某一参数值加起来,且该值必须被赋值。默认值为None
  2. default - 默认值,当参数未在命令行中出现并且也不存在于命名空间对象时所使用的值。

optional argument在option name在命令行中未出现时使用:

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', default=42)
>>> parser.parse_args(['--foo', '2'])
Namespace(foo='2')
>>> parser.parse_args([])
Namespace(foo=42)


如果目标namespace已经有attribute set了,就不会用:

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', default=42)
>>> parser.parse_args([], namespace=argparse.Namespace(foo=101))
Namespace(foo=101)


如果default值是字符串,parser就会跟命令行一样传参,比如自动根据type值进行转换;否则就使用原object:

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--length', default='10', type=int)
>>> parser.add_argument('--width', default=10.5, type=int)
>>> parser.parse_args()
Namespace(length=10, width=10.5)


  1. type - 命令行参数应当被转换成的类型。

可以是任何一个callable、入参为单一str的函数,如float、int、ascii、ord、open、argparse.FileType('w', encoding='latin-1')、pathlib.Path,也可以是用户自定义的函数。

只能返回ArgumentTypeError、TypeError或ValueError。

不建议使用bool,因为它仅会置空字符串为False,非空字符串为True。

(需要注意的是,如果直接不输入参数或者以空格作为参数,会报错error: argument --argument_name: expected one argument;如果输入""或''为参数,才会转换为False。直接写None也不会(因为直接识别的是字符串'None'))

只建议做简单处理。有些问题是报错信息太复杂。FileType的问题是不会自动close文件。


  1. help - 一个对此参数作用的简单描述。

在命令行中通过-h或-help参数查看。


  1. metavar - 在usage messages中显示的参数名。

在ArgumentParser生成的help信息中需要区别各项参数,默认用dest值作为object名。positional argument actions直接用dest值,optional argument actions大写dest值,metavar则是可选名。metavar仅改变展示时的名字,属性名仍然是dest值。不同取值的nargs可能使metavar多次被使用。

简单示例:

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', metavar='YYY')
>>> parser.add_argument('bar', metavar='XXX')
>>> parser.parse_args('X --foo Y'.split())
Namespace(bar='X', foo='Y')
>>> parser.print_help()
usage:  [-h] [--foo YYY] XXX
positional arguments:
 XXX
options:
 -h, --help  show this help message and exit
 --foo YYY


用元组传入多个metavar的实例:

>>> parser = argparse.ArgumentParser(prog='PROG')
>>> parser.add_argument('-x', nargs=2)
>>> parser.add_argument('--foo', nargs=2, metavar=('bar', 'baz'))
>>> parser.print_help()
usage: PROG [-h] [-x X X] [--foo bar baz]
options:
 -h, --help     show this help message and exit
 -x X X
 --foo bar baz


  1. dest - parse_args() 返回对象的属性名。

默认值:

positional argument actions就是add_argument()的第一个入参。

示例:

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('bar')
>>> parser.parse_args(['XXX'])
Namespace(bar='XXX')


optional argument actions是从option strings中推理得出的:从第一个开始挑一个可以开头有--的option string然后去掉--,如果没有就换成-。然后将option string中间的-全部改成_

示例:


>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('-f', '--foo-bar', '--foo')
>>> parser.add_argument('-x', '-y')
>>> parser.parse_args('-f 1 -x 2'.split())
Namespace(foo_bar='1', x='2')
>>> parser.parse_args('--foo 1 -y 2'.split())
Namespace(foo_bar='1', x='2')


指定dest入参值:

示例:


>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', dest='bar')
>>> parser.parse_args('--foo XXX'.split())
Namespace(bar='XXX')
相关实践学习
部署Stable Diffusion玩转AI绘画(GPU云服务器)
本实验通过在ECS上从零开始部署Stable Diffusion来进行AI绘画创作,开启AIGC盲盒。
目录
打赏
0
0
0
0
20
分享
相关文章
[oeasy]python064_命令行工作流的总结_vim_shell_python
本文总结了命令行工作流中的关键工具和操作,包括vim、shell和Python。主要内容如下: 1. **上次回顾**:完成了输入输出的代码编写,并再次练习了vim的使用。 2. **shell基础**:介绍了shell环境及其基本命令,如`pwd`、`cd`、`ll -l`等。 3. **Python游乐场**:通过`python3`命令进入Python交互环境,可以进行简单计算和函数调用,常用函数有`help`、`ord`、`chr`等。 4. **vim编辑器**:详细讲解了vim的三种模式(正常模式、插入模式、底行命令模式)及其切换方法,以及常用的底行命令如`:w`、`:q`、`
80 15
|
4月前
|
【10月更文挑战第2天】「Mac上学Python 2」入门篇2 - 开发环境命令行操作与文件管理
本篇将详细介绍Windows和Mac系统中的常用命令行操作与文件管理,帮助用户掌握如何通过终端或命令提示符进行文件管理和操作开发环境。内容涵盖路径切换、文件与文件夹的创建、删除、查看文件内容等基本操作,这些技能是后续Python开发的基础。
156 6
【10月更文挑战第2天】「Mac上学Python 2」入门篇2 - 开发环境命令行操作与文件管理
python方法,传参20220101 计算与当前时间差
python方法,传参20220101 计算与当前时间差
30天拿下Python之argparse模块
30天拿下Python之argparse模块
39 0
python 命令行参数 argparse详解
这篇文章是关于Python命令行参数解析库`argparse`的详细解释和使用教程。文中提供了完整的代码模板,包括如何定义命令行参数、如何添加位置参数、可选参数和布尔参数,以及如何解析和使用这些参数。文章还包含了示例代码和测试用例,以展示如何在实际程序中应用`argparse`库。
73 0
|
6月前
|
Python colorama 设置控制台、命令行输出彩色文字
Python colorama 设置控制台、命令行输出彩色文字
105 0
【python】python指南(十四):**操作符解包字典传参
【python】python指南(十四):**操作符解包字典传参
56 0
【python】python指南(二):命令行参数解析器ArgumentParser
【python】python指南(二):命令行参数解析器ArgumentParser
63 0
Day 2/100:命令行下执行python代码!
Day 2/100:命令行下执行python代码!
46 0

热门文章

最新文章