argparse 只接受几个选项 - Python

在Python中，argparse是一个非常有用的模块，可以帮助你在命令行传递参数。它提供了一种简单的方法来定义、解析和使用命令行参数。

安装Python argparse

argparse是Python自带的标准模块，所以你不需要额外安装它。只需要在你的脚本中引入即可开始使用。

import argparse

使用argparse

argparse有三个步骤：

1. 定义参数

调用argparse.ArgumentParser()来创建一个 ArgumentParser 对象，然后使用 add_argument() 方法来定义命令行参数。

parser = argparse.ArgumentParser()
parser.add_argument("--input", help="input file")

上面的例子，我们可以通过在命令行输入 --input 来传递参数。

2. 解析参数

调用 ArgumentParser 对象的 parse_args() 方法来解析参数。这将返回一个命名空间对象，其中包含所有已传递的参数。

args = parser.parse_args()

如果你使用了上面的例子，解析过程就可以通过如下方式进行：

python script.py --input filename.txt

3. 使用参数

你可以通过使用点号访问已解析的命令行参数的名称来使用它们。

print(args.input)

这将输出：

filename.txt

选项限制

有时你可能想限制命令行参数的选择。这可以通过使用 choices 参数来实现。

parser.add_argument("--output", choices=["json", "xml", "csv"])

上面的代码将只允许 --output 参数被设置为 'json'、'xml' 或 'csv' 中的一种的字符串选项。

python script.py --output json

总结

argparse是一个非常有用的模块，它可以帮助你进行命令行参数传递，并且使你的代码更易于维护和分享。通过一些简单的步骤即可开始使用argparse，希望这篇介绍对你有所帮助！