百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术资源 > 正文

Python | 使用argparse解析命令行参数

off999 2024-10-26 12:02 26 浏览 0 评论

今天是Python专题第27篇文章,我们来聊聊Python当中的命令行参数工具argparse。

命令行参数工具是我们非常常用的工具,比如当我们做实验希望调节参数的时候,如果参数都是通过硬编码写在代码当中的话,我们每次修改参数都需要修改对应的代码和逻辑显然这不太方便。比较好的办法就是把必要的参数设置成通过命令行传入的形式,这样我们只需要在运行的时候修改参数就可以了。

sys.argv

解析命令行传入参数最简单的办法就是通过sys.argv,sys.argv可以获取到我们通过命令行传入的参数。

import sys

print(sys.argv)

用法很简单,只需要调用sys.argv即可。argv是一个数组,如果参数有多个,我们可以通过下标进行访问。但是有一点需要注意,argv当中存储的结果是从Python调用开始的。

我们来看一个例子,我们随意传入一些参数,print sys.argv之后是这样的。

python test.py -a -c -d=222 
>>> ['test.py', '-a', '-c', '-d=222']

也就是说我们python运行test.py这个文件名也当做参数之一,所以我们要获取自定义参数的话需要从argv[1]开始。

sys.argv的好处是方便,我们只需要访问它就可以拿到传入的参数了。但是缺点也很明显,就是功能太少了。假如我们是看其他大神的代码,我们想要知道运行的时候需要传入什么参数,以及每个参数代表什么含义就做不到了。

为了解决这个问题,我们需要使用封装更多功能的工具,也是本篇文章的核心——argparse

基本用法

argparse是Python当中的一个库,我们需要先import一下,这个库我没记错应该是Python自带的,也不需要安装,我们直接就可以使用。

在我们使用之前,我们需要先初始化这个parse,也就是一个参数解析器。

# 这里ArgumentParser可以传入一个字符串,表示用途
parser = argparse.ArgumentParser()
parser.parse_args()

这个时候其实就已经有了一个解析器了,我们在运行的时候可以传入参数-h,表示help,也就是查看目前解析器当中定义的参数。由于我们现在什么也没有,所以能显示出来的就只有help。

必选参数

首先我们来介绍必选参数,它的定义和函数当中的必填参数是一样的,也就是说我们运行程序必须要的参数。如果不传,那么程序不应该执行会进行报错并提示。

定义必选参数的方法非常简单,我们只需要通过add_argument传入参数的名称就可以了。

import argparse

parser = argparse.ArgumentParser("For test the parser")
parser.add_argument('test')
args = parser.parse_args()

print(args.test)

这样我们就定义了一个名叫test的参数,我们可以通过args.test来访问它。

这个时候我们再运行python test.py -h就会发现提示的信息当中多了一行:

告诉我们必选参数当中有test,必选参数直接传入,不需要加上前缀。所以我们执行的时候直接python test.py xxx就可以了。

可选参数

有必选参数当然就有可选参数,可选参数由于可选可不选, 所以我们在使用的时候需要在参数前加上标识-或者--。比如我们参数名叫做test,可以定义成-test或者--test,这两种都可以,也可以这两种都定义。

parser.add_argument('-test', '--test')

我们运行-h可以发现optional arguments当中多了test和--test。

但是这个只print出来了参数名,并没有告诉我们这个参数究竟是做什么的,像是help参数后面就跟了show this help message and exit这个提示语。如果我们也希望help能够提示我们参数的作用怎么办呢?

我们可以通过help参数传入我们希望打印出来的提示语,这样方便使用者在使用的时候了解参数的情况。

比如我们把这行语句改成:

parser.add_argument('-test', '--test', help='just for help')

这样当我们运行的时候,就会看到提示语了:

默认值

如果参数很多的时候,我们有时候可能不希望每一个都指定一个值,而是希望可以在不填的时候有一个默认值。这个想法非常正常,想要做到这点也很简单,我们可以通过default参数来指定。

import argparse

parser = argparse.ArgumentParser("For test the parser")
parser.add_argument('-test', '--test', default=1, help='just for help')
args = parser.parse_args()

print(args.test)

比如这样我们在代码当中把test参数的默认值设置成了1,当我们运行的时候,如果不填test这个参数的话,那么程序就会使用它的默认值也就是1。

但有一点默认值的信息并不会print在help当中,所以我们需要自己在提示语当中告知使用者默认值是多少。

type

我们可以定义参数的默认值,当然也可以定义它的类型。

因为命令行传入的参数默认都是字符串,如果我们要进行数学上的计算,使用str还需要自己转换,这就很不方便。我们可以在传入参数的时候就完成类型的匹配,这样如果传入参数的类型不对, 那么直接报错,不往下运行。

想要做到这点也很简单,通过type参数就可以实现。

parser.add_argument('-test', '--test', default=1, type=int, help='just for help')

比如当我们定义了一个int型的参数,而传入的是类型不匹配的话,那么就会引起报错:

报错信息当中写得很清楚,我们得到了一个无效的int的值,它是abc。

可选值

它同样还支持可选值,可选值很好理解,就是我们希望限定传入参数的范围仅仅在几个值当中。比如说我们希望传入的值不是0就是1,或者是在某几个具体的值当中,这个时候我们可以通过choices参数来实现这一点。

choices参数传入的是一个list,也就是我们的限定范围,只有在这个范围当中的值才被允许。

parser.add_argument('-test', '--test', default=1, choices=[2, 3, 4], type=int, help='just for help')

如果我们运行传入test=1,那么就会引起报错,告诉我们传入的值不在choices范围当中。

这是一个挺有意思的例子,仔细看会发现我们默认值设置成了1,但是可选值当中并没有1。这也是允许的,默认值可以不在可选值范围内,但是当我们传入1就会触发可选值校验。

action

action是一个很神奇也很有用的操作,可以指定参数的处理方式。我们默认的方式是store,也就是存储的意思,这个我们都能理解。除此之外,还有store_true,它表示出现则是true,否则是false。

parser.add_argument('-test', '--test', action='store_true', help='just for help')

当我们把test参数的定义改成这样之后,我们来对比一下运行的结果就明白了。

除了store_true之外还有store_const,也就是说出现就指定为一个固定值。

parser.add_argument('-test', '--test', action='store_const', const=23, help='just for help')

这样当我们指定-test参数之后,它会自动被赋值成23。

除了这两个之外,另外一个很常用的参数是append,可以将多次出现的同一个参数自动存入一个list当中。

parser.add_argument('-test', '--test', action='append', type=int, help='just for help')

nargs

nargs也是一个非常有用的参数,可以对参数进行一些花式操作。nargs的传入参数有以下几种,首先是N,也就是一个整数。代表可以接收N个参数值,这N个值会被存入一个list当中。

parser.add_argument('-test', '--test', nargs=2, type=int, help='just for help')

另外一种传入的参数是'+'或者是'*',它可以将任意多个值存入一个list当中。

parser.add_argument('-test', '--test', nargs='*', type=int, help='just for help')

总结

有了parser之后,我们在Python当中处理命令行参数会变得非常简单,我们可以做各种各样的定制化操作。除了我们上面介绍的之外,还有一些其他的做法,相对来说不是非常常用,所以就不一一穷尽了,感兴趣的同学可以自行了解一下。

今天的文章到这里就结束了,如果喜欢本文的话,请来一波素质三连,给我一点支持吧(关注、转发、点赞)。

- END -

本文始发于公众号:TechFlow,求个关注

相关推荐

面试官:来,讲一下枚举类型在开发时中实际应用场景!

一.基本介绍枚举是JDK1.5新增的数据类型,使用枚举我们可以很好的描述一些特定的业务场景,比如一年中的春、夏、秋、冬,还有每周的周一到周天,还有各种颜色,以及可以用它来描述一些状态信息,比如错...

一日一技:11个基本Python技巧和窍门

1.两个数字的交换.x,y=10,20print(x,y)x,y=y,xprint(x,y)输出:102020102.Python字符串取反a="Ge...

Python Enum 技巧,让代码更简洁、更安全、更易维护

如果你是一名Python开发人员,你很可能使用过enum.Enum来创建可读性和可维护性代码。今天发现一个强大的技巧,可以让Enum的境界更进一层,这个技巧不仅能提高可读性,还能以最小的代价增...

Python元组编程指导教程(python元组的概念)

1.元组基础概念1.1什么是元组元组(Tuple)是Python中一种不可变的序列类型,用于存储多个有序的元素。元组与列表(list)类似,但元组一旦创建就不能修改(不可变),这使得元组在某些场景...

你可能不知道的实用 Python 功能(python有哪些用)

1.超越文件处理的内容管理器大多数开发人员都熟悉使用with语句进行文件操作:withopen('file.txt','r')asfile:co...

Python 2至3.13新特性总结(python 3.10新特性)

以下是Python2到Python3.13的主要新特性总结,按版本分类整理:Python2到Python3的重大变化Python3是一个不向后兼容的版本,主要改进包括:pri...

Python中for循环访问索引值的方法

技术背景在Python编程中,我们经常需要在循环中访问元素的索引值。例如,在处理列表、元组等可迭代对象时,除了要获取元素本身,还需要知道元素的位置。Python提供了多种方式来实现这一需求,下面将详细...

Python enumerate核心应用解析:索引遍历的高效实践方案

喜欢的条友记得关注、点赞、转发、收藏,你们的支持就是我最大的动力源泉。根据GitHub代码分析统计,使用enumerate替代range(len())写法可减少38%的索引错误概率。本文通过12个生产...

Python入门到脱坑经典案例—列表去重

列表去重是Python编程中常见的操作,下面我将介绍多种实现列表去重的方法,从基础到进阶,帮助初学者全面掌握这一技能。方法一:使用集合(set)去重(最简单)pythondefremove_dupl...

Python枚举类工程实践:常量管理的标准化解决方案

本文通过7个生产案例,系统解析枚举类在工程实践中的应用,覆盖状态管理、配置选项、错误代码等场景,适用于Web服务开发、自动化测试及系统集成领域。一、基础概念与语法演进1.1传统常量与枚举类对比#传...

让Python枚举更强大!教你玩转Enum扩展

为什么你需要关注Enum?在日常开发中,你是否经常遇到这样的代码?ifstatus==1:print("开始处理")elifstatus==2:pri...

Python枚举(Enum)技巧,你值得了解

枚举(Enum)提供了更清晰、结构化的方式来定义常量。通过为枚举添加行为、自动分配值和存储额外数据,可以提升代码的可读性、可维护性,并与数据库结合使用时,使用字符串代替数字能简化调试和查询。Pytho...

78行Python代码帮你复现微信撤回消息!

来源:悟空智能科技本文约700字,建议阅读5分钟。本文基于python的微信开源库itchat,教你如何收集私聊撤回的信息。[导读]Python曾经对我说:"时日不多,赶紧用Python"。于是看...

登录人人都是产品经理即可获得以下权益

文章介绍如何利用Cursor自动开发Playwright网页自动化脚本,实现从选题、写文、生图的全流程自动化,并将其打包成API供工作流调用,提高工作效率。虽然我前面文章介绍了很多AI工作流,但它们...

Python常用小知识-第二弹(python常用方法总结)

一、Python中使用JsonPath提取字典中的值JsonPath是解析Json字符串用的,如果有一个多层嵌套的复杂字典,想要根据key和下标来批量提取value,这是比较困难的,使用jsonpat...

取消回复欢迎 发表评论: