python如何从命令行输入参数

Python可以通过sys模块、argparse模块、以及optparse模块来从命令行输入参数、方便传递参数、灵活处理输入。 其中，argparse模块功能最强大、最常用。以下将详细描述使用argparse模块从命令行获取参数的方法。

在Python中，从命令行输入参数是一项非常有用的技能，尤其在编写脚本和自动化任务时。通过从命令行传递参数，可以使脚本更加灵活和动态。以下是如何使用argparse模块从命令行输入参数的详细步骤。

一、理解argparse模块

argparse是Python标准库中的一个模块，专门用来解析命令行参数。它提供了一个易于使用的接口来定义脚本需要的参数，并自动生成帮助和使用信息。

1、安装和导入argparse模块

尽管argparse是标准库的一部分，无需额外安装，但我们仍需导入它：

import argparse

二、创建ArgumentParser对象

ArgumentParser对象是argparse模块的核心，用于处理命令行参数。我们首先需要创建一个ArgumentParser对象：

parser = argparse.ArgumentParser(description='This is a description of the script.')

1、description参数

description参数用于提供对脚本的简要描述，这些描述会在用户请求帮助时显示。

三、添加参数

接下来，我们需要定义脚本所需的参数，并将其添加到ArgumentParser对象中。我们使用add_argument方法来实现这一点：

parser.add_argument('input', type=str, help='Input file name')
parser.add_argument('output', type=str, help='Output file name')
parser.add_argument('-v', '--verbose', action='store_true', help='Increase output verbosity')

1、位置参数

位置参数是必须提供的参数，在add_argument方法中不需要指定任何前缀字符。例如，上面的input和output参数。

2、可选参数

可选参数是可以省略的参数，通常以短横线-或双短横线--开头。例如，上面的-v和--verbose参数。

3、参数类型和帮助信息

在定义参数时，可以指定参数的类型（如str、int等）和帮助信息，这些帮助信息会在用户请求帮助时显示。

四、解析参数

一旦定义了所有需要的参数，我们就可以解析命令行参数了。使用parse_args方法来解析参数：

args = parser.parse_args()

五、使用参数

解析后的参数存储在args对象中，可以像访问对象属性一样访问这些参数：

input_file = args.input
output_file = args.output
if args.verbose:
    print('Verbose mode is enabled.')

六、完整示例

以下是一个完整的示例，演示如何从命令行输入参数：

import argparse
def main():
    # 创建ArgumentParser对象
    parser = argparse.ArgumentParser(description='Process some files.')
    # 添加参数
    parser.add_argument('input', type=str, help='Input file name')
    parser.add_argument('output', type=str, help='Output file name')
    parser.add_argument('-v', '--verbose', action='store_true', help='Increase output verbosity')
    # 解析参数
    args = parser.parse_args()
    # 使用参数
    input_file = args.input
    output_file = args.output
    if args.verbose:
        print(f'Processing input file: {input_file}')
        print(f'Saving output to: {output_file}')
    # 这里可以添加更多的代码来处理文件
if __name__ == '__main__':
    main()

七、运行脚本

保存上述脚本为一个Python文件（例如script.py），然后在命令行中运行它：

python script.py input.txt output.txt --verbose

八、处理更多复杂的参数

argparse模块还支持处理更加复杂的参数，例如：

1、默认值

可以为参数提供默认值，当用户未指定该参数时使用默认值：

parser.add_argument('-t', '--timeout', type=int, default=30, help='Timeout value in seconds')

2、参数互斥

可以定义互斥组，确保在命令行中只能指定其中一个参数：

group = parser.add_mutually_exclusive_group()
group.add_argument('-a', '--option_a', action='store_true', help='Option A')
group.add_argument('-b', '--option_b', action='store_true', help='Option B')

3、解析子命令

可以定义子命令，以支持更复杂的命令行接口：

subparsers = parser.add_subparsers(title='subcommands', description='valid subcommands', help='additional help')
subparser_a = subparsers.add_parser('subcommand_a', help='Subcommand A')
subparser_a.add_argument('foo', type=int, help='Foo value')
subparser_b = subparsers.add_parser('subcommand_b', help='Subcommand B')
subparser_b.add_argument('bar', type=str, help='Bar value')