如何使用Python argparse定义逗号分隔的双整数格式参数类型
实现方式
argparse 没有预置专门匹配「整数+逗号+整数」格式的参数类型,但官方支持通过自定义type函数的方式实现需求,不需要额外手动做后置校验,这是官方推荐的标准方案。
实现步骤
- 先定义参数校验解析函数
把格式校验、类型转换的逻辑都封装在函数里,不符合要求时抛出argparse.ArgumentTypeError即可,argparse会自动处理错误提示和程序退出逻辑。
import argparse def point_type(val): parts = val.split(',') # 校验是否为两段内容 if len(parts) != 2: raise argparse.ArgumentTypeError("格式错误,必须为「整数,整数」形式,示例:0,0") try: # 去除两端空格后转整数 x = int(parts[0].strip()) y = int(parts[1].strip()) except ValueError: raise argparse.ArgumentTypeError("逗号两侧必须为合法整数") # 直接返回解析后的整数元组,后续使用不需要再转换 return (x, y)
- 修改参数注册逻辑
把add_argument的type参数替换为上面定义的point_type即可:
parser = argparse.ArgumentParser() parser.add_argument('--point', type=point_type, metavar='X,Y', required=True) args = parser.parse_args() # 可直接使用解析好的坐标元组 print(f"x坐标:{args.point[0]}, y坐标:{args.point[1]}")
效果说明
- 传入合法参数
python main.py --point 3,5,程序可直接拿到(3,5)格式的整数元组 - 传入非法参数(比如
--point 1、--point a,2、--point 1,2,3)时,argparse会自动输出你定义的错误提示并终止程序,不需要额外写判断逻辑。
内容的提问来源于stack exchange,提问作者usersman
相关产品推荐
相关产品推荐

