argparse模块通过创建argumentparser对象处理命令行参数。首先导入模块并创建解析器,使用add_argument()定义参数类型,如位置参数parser.add_argument(“echo”)和可选参数parser.add_argument(“–verbose”)。其次区分参数形式,位置参数无前缀且必填,如python script.py hello;可选参数带–或-前缀且非必须,如–verbose或-v。接着设置参数类型、默认值与选项,如type=int指定整型,default=”result.txt”设定默认输出文件名,choices=[“fast”, “slow”]限制选项范围。最后支持多值或多现参数,如nargs=”+”接受多个数值,action=”count“统计出现次数。掌握这些基本用法后,可根据需求逐步深入使用。
在写 Python 脚本时,经常需要处理命令行参数。argparse 模块就是用来做这件事的标准库之一,用起来简单又灵活。下面我们就来聊聊它的基本用法和一些实用技巧。
如何开始使用 argparse
首先,你需要导入 argparse 模块,并创建一个 ArgumentParser 对象。这个对象会帮你解析命令行参数。
import argparse parser = argparse.ArgumentParser(description='这是一个简单的示例')
这里的关键是 add_argument() 方法,它用来定义你希望接收的参数。比如:
立即学习“Python免费学习笔记(深入)”;
-
添加一个位置参数(必须提供的参数):
parser.add_argument("echo", help="打印你输入的内容")
-
添加一个可选参数:
parser.add_argument("--verbose", help="启用详细输出模式", action="store_true")
然后调用 parse_args() 就能获取用户输入的参数了:
args = parser.parse_args() print(args.echo) if args.verbose: print("详细模式已开启")
怎么区分位置参数和可选参数
位置参数是你运行脚本时必须输入的参数,顺序固定。比如:
python script.py hello
这里的 hello 就是一个位置参数。
可选参数则是带 — 或 – 前缀的参数,不是必须的。比如:
python script.py --verbose hello
还可以给可选参数加短格式,比如:
parser.add_argument("-v", "--verbose", help="启用详细模式", action="store_true")
这样用户既可以写 –verbose,也可以简写成 -v。
参数类型与默认值怎么设置
argparse 默认把参数当作字符串处理,但你可以指定类型:
parser.add_argument("square", type=int, help="计算平方数")
如果你希望某个参数有默认值,可以这样写:
parser.add_argument("--output", default="result.txt", help="输出文件名")
这样即使用户不提供 –output,程序也能用默认值继续执行。
还有些时候,你想让参数只能从几个选项中选,可以用 choices:
parser.add_argument("--mode", choices=["fast", "slow"], default="fast")
处理多个参数或重复参数
如果你想支持多个值的参数,比如让用户输入多个数字:
parser.add_argument("--numbers", nargs="+", type=int)
这样就能接受像 –numbers 1 2 3 这样的输入。
如果参数可以出现多次,比如日志级别可以多次传入来增加等级:
parser.add_argument("-v", "--verbose", action="count", default=0)
那么 -v 是一次,-vv 是两次,适合控制日志级别。
基本上就这些。argparse 的功能很全,用的时候不需要一开始就掌握所有特性,先学会最常用的几种方式,遇到复杂需求再查文档也不迟。关键是理解参数类型、可选参数、默认值这些概念,剩下的就是多练多用。
评论(已关闭)
评论已关闭