使用 argparse 库添加命令行参数

在 Python 中，argparse 库是用于处理命令行参数和选项的标准模块。它可以让我们轻松地编写带有命令行接口的 Python 程序。 parser.add_argument 方法是使用 argparse 命令行解析器时的一个重要函数，它用于定义可以使用的命令行参数和选项。

示例

下面是一个简单的示例，通过 parser.add_argument 添加了两个命令行参数：

import argparse
 
parser = argparse.ArgumentParser()
parser.add_argument("--input", help="输入文件路径")
parser.add_argument("--output", help="输出文件路径")
args = parser.parse_args()
print(args.input)
print(args.output)

在上面的代码中，我们使用了 argparse 库中的 ArgumentParser() 函数定义了一个解析器对象。然后，我们使用 parser.add_argument() 方法添加了两个命令行参数。我们可以在命令行中使用 --input 和 --output 参数来指定输入和输出文件路径。

最后，我们使用 parser.parse_args() 方法来解析命令行参数，并将其存储在 args 对象中。然后我们可以使用 args.input 和 args.output 来访问这些参数。

我们可以通过这样的方式在命令行中运行这个脚本：

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

然后，这个脚本将 “input.txt” 文件中的文本读取到内存中，并将其写入 “output.txt” 文件。

参数类型

parser.add_argument 方法还可以用于指定参数的类型。例如，下面的代码添加了一个类型为整数的命令行参数：

parser.add_argument("--number", help="一个整数", type=int)

这样，我们在命令行中输入一个整数时，argparse 库会自动将其解析为整数类型。

参数数组

有时，我们可能需要在命令行中传递数组。我们可以使用 nargs 参数来指定一个参数接受多个值。例如，下面的代码将“--numbers”参数设置为接受多个整数：

parser.add_argument("--numbers", help="多个整数", nargs="+", type=int)

这将让我们在命令行中输入多个整数，并将它们作为 Python 列表存储在 args.numbers 变量中。例如，我们可以这样调用这个脚本：

$ python script.py --numbers 1 2 3 4 5

然后 args.numbers 将包含 [1, 2, 3, 4, 5]。

总结

在 Python 中，使用 argparse 库可以轻松地添加命令行参数和选项。parser.add_argument 是一个有用的函数，它用于指定可以使用的命令行参数和选项的类型。我们可以使用 type 参数指定参数的类型，使用 nargs 参数指定参数接受多个值。