Python sys.argv用法详解

小蓝 • 2024-12-04 10:23 • 编程

一、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，我们可以在实际开发中方便地处理参数，提高脚本的可读性和可维护性。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/198244.html