Python 提供了多种模块来解析命令行参数，其中最常用的是 argparse 模块，它功能强大且易于使用，能够帮助开发者快速构建用户友好的命令行接口。sys 模块和 getopt 模块也可以用于处理命令行参数，但 argparse 因其丰富的功能和灵活的配置而成为首选。
使用 argparse 模块解析命令行参数
argparse 模块是 Python 标准库的一部分，支持自动生成帮助信息、参数类型检查、默认值设置以及子命令等功能，以下是使用 argparse 的基本步骤：
-
导入模块：首先需要导入
argparse模块。import argparse
-
创建解析器对象：使用
ArgumentParser类创建一个解析器对象，可以设置程序描述、帮助前缀等信息。parser = argparse.ArgumentParser(description='这是一个示例程序')
-
添加参数：通过
add_argument方法添加需要解析的参数，可以指定参数名称、类型、默认值、帮助信息等。
（图片来源网络，侵删）parser.add_argument('--name', type=str, default='Guest', help='指定用户名') parser.add_argument('--age', type=int, help='指定年龄') parser.add_argument('--verbose', action='store_true', help='启用详细输出')--name是一个可选参数，类型为字符串，默认值为 'Guest'。--age是一个可选参数，类型为整数,无默认值。--verbose是一个布尔标志参数，当指定时值为True，否则为False。
-
解析参数：使用
parse_args方法解析命令行参数，返回一个命名空间对象,包含所有参数的值。args = parser.parse_args() print(f'Name: {args.name}') print(f'Age: {args.age}') print(f'Verbose: {args.verbose}') -
运行程序：通过命令行运行程序并传递参数，
python script.py --name Alice --age 25 --verbose
输出结果为：
Name: Alice Age: 25 Verbose: True
高级功能
argparse 还支持更复杂的功能，如子命令、互斥参数、可选参数组等,添加子命令：
subparsers = parser.add_subparsers(dest='command')
parser_add = subparsers.add_parser('add', help='添加操作')
parser_add.add_argument('x', type=int, help='第一个数')
parser_add.add_argument('y', type=int, help='第二个数')
args = parser.parse_args()
if args.command == 'add':
print(f'结果: {args.x + args.y}')
使用 sys 模块解析参数
sys 模块的 sys.argv 列表可以直接获取命令行参数，sys.argv[0] 是脚本名称,后续元素是参数。
import sys
print('脚本名称:', sys.argv[0])
print('参数列表:', sys.argv[1:])
运行 python script.py arg1 arg2,输出：
脚本名称: script.py
参数列表: ['arg1', 'arg2']但 sys.argv 需要手动处理参数类型和逻辑,适合简单的场景。
使用 getopt 模块解析参数
getopt 模块提供了类似 Unix 风格的参数解析功能，支持短选项（如 -v）和长选项（如 --verbose）。
import getopt
import sys
try:
opts, args = getopt.getopt(sys.argv[1:], 'h:v', ['help', 'verbose='])
except getopt.GetoptError:
print('参数错误')
sys.exit(2)
for opt, arg in opts:
if opt in ('-h', '--help'):
print('帮助信息')
elif opt in ('-v', '--verbose'):
print(f'详细模式: {arg}')
参数解析对比
以下是三种方法的对比：
| 模块 | 优点 | 缺点 | 适用场景 |
|---|---|---|---|
argparse |
功能强大，自动生成帮助信息 | 代码量稍多 | 复杂命令行程序 |
sys |
简单直接，无需额外配置 | 需手动处理参数逻辑 | 简单脚本或快速原型 |
getopt |
支持 Unix 风格参数 | 功能有限，错误处理较复杂 | 传统 Unix 风格程序 |
相关问答 FAQs
Q1: 如何在 argparse 中设置互斥参数？
A1: 使用 add_mutually_exclusive_group 方法创建互斥参数组,确保同一时间只能选择其中一个参数。
group = parser.add_mutually_exclusive_group()
group.add_argument('--verbose', action='store_true')
group.add_argument('--quiet', action='store_true')
Q2: 如何在 argparse 中处理可选参数和位置参数的组合？
A2: 可选参数以 开头,位置参数直接指定名称。
parser.add_argument('--input', type=str, help='输入文件路径')
parser.add_argument('output', type=str, help='输出文件路径')
运行时需提供位置参数 output，可选参数 input 可选。
python script.py output.txt --input input.txt
