如何在 Python 脚本中使用 argparse 模块处理命令行参数

159 阅读1分钟

在 Python 中,你可以使用 argparse 模块来处理命令行参数。通过这个模块,你可以轻松解析从命令行传递的参数并在脚本中使用它们。以下是如何实现的示例:

1. 创建一个 test.py 脚本

import argparse

# 创建解析器对象
parser = argparse.ArgumentParser(description="处理命令行参数示例")

# 添加命令行参数
parser.add_argument("-name", type=str, help="输入名字")
parser.add_argument("-sex", type=str, help="输入性别")

# 解析命令行参数
args = parser.parse_args()

# 获取参数值
name = args.name
sex = args.sex

# 输出结果
print(f"名字: {name}")
print(f"性别: {sex}")

2. 运行脚本

在终端或命令行中运行以下命令:

python test.py -name zhangsan -sex man

3. 脚本输出

执行上述命令后,脚本将输出以下内容:

名字: zhangsan
性别: man

解释

  • argparse.ArgumentParser():创建一个解析器对象,用于管理命令行参数的解析。
  • parser.add_argument("-name", type=str, help="输入名字"):定义了一个 -name 参数,并指定它的类型为 str(字符串)。
  • args = parser.parse_args():解析所有传入的命令行参数,并将其存储在 args 对象中。
  • args.nameargs.sex:用于访问命令行中传递的 -name-sex 参数的值。

通过这种方式,你可以在运行 python test.py 时传递变量,并在脚本中轻松地访问这些变量。