一、sys.argv基础入门
sys.argv是Python中sys模块中的一项功能,用于处理命令行参数,以列表形式返回用户输入的所有命令行参数。
在命令行中,我们可以输入类似于 python test.py arg1 arg2 arg3 这样的命令。其中test.py就是我们要执行的脚本文件,后面的arg1,arg2,arg3是我们传递给脚本的参数。sys.argv就是用来获取这些参数的。
import sys print(sys.argv)
输出结果就是一个列表,第一项是脚本的名称,从第二项开始依次是我们传递的参数:
['test.py', 'arg1', 'arg2', 'arg3']
二、sys.argv高级应用
除了上面的基础用法,我们还可以根据实际需求,在sys.argv的基础上进行一些高级应用。
1. 参数个数判断
通常情况下,我们需要判断用户输入的参数个数是否符合脚本的要求。这时候,我们可以使用len(sys.argv)进行参数个数的判断:
import sys if len(sys.argv) != 3: print("Usage: python test.py arg1 arg2")
上面的例子中,我们要求脚本必须传入两个参数,否则就输出用法提示。
2. 参数类型转换
sys.argv获取的是一个字符串列表,如果想将其中的参数转换成整数、浮点数等类型,可以使用Python中的内置函数进行转换。
import sys ad_num = int(sys.argv[1]) if ad_num == 1: print("Loading ad 1 now.")
上面的例子中,第一个参数被转换为整型,用于判断加载广告1。
3. 参数解析
如果脚本需要支持更加复杂的参数,比如选项、短选项等,那么就需要用到Python中的argparse模块,它可以帮助我们方便地解析参数,并且支持自动生成帮助文档等功能。
import argparse import sys parser = argparse.ArgumentParser(description='Process some integers.') 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)') args = parser.parse_args() print(args.accumulate(args.integers))
上面的例子中,我们首先引入了argparse模块,并且定义了一个ArgumentParser实例,其中description是帮助信息。接着我们通过add_argument()方法定义了参数的名称、类型、个数、默认值等等。最后通过parse_args()方法解析用户传入的参数。
三、总结
以上就是Python的sys.argv用法的全部内容了。通过 sys.argv,我们可以在实际开发中方便地处理参数,提高脚本的可读性和可维护性。